Searched defs:tensor_spec (Results 1 – 8 of 8) 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() local87 OpSignatureTensorSpec tensor_spec = {kTfLiteNoType}; in GetOpSignatureTensorSpecs() local
32 OpSignatureTensorSpec tensor_spec = {}; in CreateOpSignatureTensorSpecs() local43 OpSignatureTensorSpec tensor_spec = {}; in CreateOpSignatureTensorSpecs() local54 OpSignatureTensorSpec tensor_spec = {}; in CreateOpSignatureTensorSpecs() local
54 TensorSpec tensor_spec; in TEST() local
134 TaggedValue tensor_spec(shape, dtype); in ImportModule() local
51 tensor_spec = LazyLoader( variable
40 tensor_spec = LazyLoader( variable
2568 def _create_distributed_tensor_spec(strategy, tensor_spec): argument