deepr.jobs.ExportXlaModelMetadata

class deepr.jobs.ExportXlaModelMetadata(path_optimized_model, path_metadata, graph_name, metadata_name, feed_shapes, fetch_shapes)[source]

Export xla compatible model metadata from a saved model

path_optimized_model

Path to directory containing optimized saved model exports to convert

Type:

str

path_metadata

Path to directory that will contain the metadata

Type:

str

graph_name

Name of the saved model graph (name of the protobuf file)

Type:

str

metadata_name

Name of the metadata file

Type:

str

feed_shapes

Shapes of feeds to expose

Type:

Dict[str, List[int]]

fetch_shapes

Shapes of fetches to expose

Type:

Dict[str, List[int]]

__init__(path_optimized_model, path_metadata, graph_name, metadata_name, feed_shapes, fetch_shapes)

Methods

__init__(path_optimized_model, ...)

run()

Run Job

Attributes

path_optimized_model

path_metadata

graph_name

metadata_name

feed_shapes

fetch_shapes