models.meta Namespace Reference

Classes
class	Model

Functions
def	model_path (name)
def	check_tf_identity (runner)
def	check_identity (runner)
def	check_identity_identity (runner)
def	check_dynamic_identity (runner, shapes)
def	check_empty_tensor_expand (runner, shapes)
def	check_reshape (runner)
def	no_check_implemented (runner)

Variables
dictionary	TF_MODELS
dictionary	ONNX_MODELS

Function Documentation

model_path()

def models.meta.model_path

(

name

)

check_tf_identity()

def models.meta.check_tf_identity

(

runner

)

check_identity()

def models.meta.check_identity

(

runner

)

check_identity_identity()

def models.meta.check_identity_identity

(

runner

)

check_dynamic_identity()

def models.meta.check_dynamic_identity	(		runner,
			shapes
	)

check_empty_tensor_expand()

def models.meta.check_empty_tensor_expand	(		runner,
			shapes
	)

check_reshape()

def models.meta.check_reshape

(

runner

)

no_check_implemented()

def models.meta.no_check_implemented

(

runner

)

Variable Documentation

TF_MODELS

dictionary models.meta.TF_MODELS

Initial value:

=  {
    "identity": Model(path=model_path("tf_identity.pb"), LoaderType=GraphFromFrozen, check_runner=check_tf_identity),
}

ONNX_MODELS

dictionary models.meta.ONNX_MODELS

Initial value:

=  {
    "identity": Model(path=model_path("identity.onnx"), LoaderType=BytesFromPath, check_runner=check_identity, input_metadata=TensorMetadata().add("x", dtype=np.float32, shape=(1, 1, 2, 2))),
    "identity_identity": Model(path=model_path("identity_identity.onnx"), LoaderType=BytesFromPath, check_runner=check_identity_identity),
    "dynamic_identity": Model(path=model_path("dynamic_identity.onnx"), LoaderType=BytesFromPath, check_runner=check_dynamic_identity, input_metadata=TensorMetadata().add("X", dtype=np.float32, shape=(1, 1, -1, -1))),
    "empty_tensor_expand": Model(path=model_path("empty_tensor_expand.onnx"), LoaderType=BytesFromPath, check_runner=check_empty_tensor_expand),
 
    "scan": Model(path=model_path("scan.onnx"), LoaderType=BytesFromPath, check_runner=no_check_implemented),
    "dim_param": Model(path=model_path("dim_param.onnx"), LoaderType=BytesFromPath, check_runner=no_check_implemented),
    "tensor_attr": Model(path=model_path("tensor_attr.onnx"), LoaderType=BytesFromPath, check_runner=no_check_implemented),
    "identity_with_initializer": Model(path=model_path("identity_with_initializer.onnx"), LoaderType=BytesFromPath, check_runner=no_check_implemented),
    "const_foldable": Model(path=model_path("const_foldable.onnx"), LoaderType=BytesFromPath, check_runner=no_check_implemented),
    "reshape": Model(path=model_path("reshape.onnx"), LoaderType=BytesFromPath, check_runner=check_reshape),
}