Searched defs:tensor_spec (Results 1 – 12 of 12) 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
80 auto tensor_spec = in TEST() local
134 TaggedValue tensor_spec(shape, dtype); in ImportModule() local
138 for (const auto& tensor_spec : in InitializeFlTask() local
48 tensor_spec = LazyLoader( variable
41 tensor_spec = LazyLoader( variable
82 tensor_spec = LazyLoader( variable
178 for (const tensorflow::TensorSpecProto& tensor_spec : in CreateComputationResults() local
2056 def _create_distributed_tensor_spec(strategy, tensor_spec): argument