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`