Searched defs:tensor_spec (Results 1 – 9 of 9) sorted by relevance
175 TaggedValue tensor_spec(unknown_shape, DT_FLOAT); in TEST_P() local198 TaggedValue tensor_spec(unknown_shape, DT_FLOAT); in TEST_P() local243 TaggedValue tensor_spec(unknown_shape, DT_FLOAT); in TEST_P() local
51 OpSignatureTensorSpec tensor_spec = {kTfLiteNoType}; in GetOpSignatureTensorSpecs() local98 OpSignatureTensorSpec tensor_spec = {kTfLiteNoType}; in GetOpSignatureTensorSpecs() local
32 OpSignatureTensorSpec tensor_spec = {}; in CreateOpSignatureTensorSpecs() local45 OpSignatureTensorSpec tensor_spec = {}; in CreateOpSignatureTensorSpecs() local59 OpSignatureTensorSpec tensor_spec = {}; in CreateOpSignatureTensorSpecs() local70 OpSignatureTensorSpec tensor_spec = {}; in CreateOpSignatureTensorSpecs() local
54 TensorSpec tensor_spec; in TEST() local
134 TaggedValue tensor_spec(shape, dtype); in ImportModule() local
48 tensor_spec = LazyLoader( variable
41 tensor_spec = LazyLoader( variable
82 tensor_spec = LazyLoader( variable
2056 def _create_distributed_tensor_spec(strategy, tensor_spec): argument