rztdl.dl package

Subpackages

Submodules

rztdl.dl.groups module

@created on: 01/20/20, @author: Prathyush SP, @version: v0.0.1

Description:

Sphinx Documentation Status:

class rztdl.dl.groups.Group(name: str, inputs: typing.List[typing.Union[str, tensorflow.python.framework.ops.Tensor]])[source]

Bases: rztdl.dl.model.RZTModel

Group Class

Parameters:
  • name (str) – Name of the group
  • inputs (List[Union[str, Tensor]]) – Inputs for the group
add(component)[source]

Add components to the group - Lazy addition (Components are validated and are stored in a list. Components are created in compile method) :type component: RZTComponent :param component: RZTComponent

close()[source]

Finalize Group :return:

rztdl.dl.model module

@created on: 12/13/19, @author: Himaprasoon, @version: v0.0.1

Description:

Sphinx Documentation Status:

class rztdl.dl.model.RZTModel(name: str, dist=False)[source]

Bases: object

Model Class

Parameters:name (str) – str
add(component)[source]

Adds a component to the graph :type component: RZTComponent :param component: :return:

add_group()[source]
clear()[source]
close()[source]
get_complete_model(outputs=None)[source]
get_component(name)[source]
get_component_by_name(name)[source]
get_component_from_output(output_name)[source]
get_component_from_tensor(tensor)[source]

Check if tensor is part of graph and gets corresponding component :type tensor: Tensor :param tensor: :return:

get_required_buffers(components, get_names=False)[source]

Return list of Input layers given a list of components :param get_names: if output needs to be a list of component_name :type components: List[Union[str, RZTComponent]] :param components: :return:

get_validation_model(component_or_output_names, get_mapping=False)[source]
predecessors(component)[source]

Gets predecessors of the component in the dag :param component: :return:

write_graph_tensorboard(*args, **kwargs)[source]
rztdl.dl.model.check_closed(f)[source]

Module contents

@created on: 2019-12-03, @author: Himaprasoon, @version: v0.0.1

Description:

Sphinx Documentation Status: