Home
last modified time | relevance | path

Searched refs:HasUnspecifiedDimension (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dslice.cc148 if (!HasUnspecifiedDimension(output) && ShapeHasRank(output->dims)) { in Prepare()
154 HasUnspecifiedDimension(input)) { in Prepare()
Dkernel_util.h326 bool HasUnspecifiedDimension(const TfLiteTensor* tensor);
Dkernel_util_test.cc904 TEST_F(KernelUtilTest, HasUnspecifiedDimension) { in TEST_F() argument
912 EXPECT_TRUE(HasUnspecifiedDimension(&tensor)); in TEST_F()
915 EXPECT_FALSE(HasUnspecifiedDimension(&tensor)); in TEST_F()
Dtopk_v2.cc218 if (IsConstantTensor(top_k) && !HasUnspecifiedDimension(input)) { in Prepare()
Dkernel_util.cc582 bool HasUnspecifiedDimension(const TfLiteTensor* tensor) { in HasUnspecifiedDimension() function
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_signature.cc114 tensor_spec.is_shape_dynamic = HasUnspecifiedDimension(tfl_tensor); in GetOpSignatureTensorSpecs()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1839 bool has_unspecified_dimensions = ::tflite::HasUnspecifiedDimension(tensor); in AddTensor()
4530 if (!HasUnspecifiedDimension(&context->tensors[tensor_index])) continue; in Init()
4882 if (!can_infer_output_shape && HasUnspecifiedDimension(&tensor)) { in Invoke()
4929 ::tflite::HasUnspecifiedDimension(tensor)) { in Invoke()
5055 ::tflite::HasUnspecifiedDimension(tensor)) { in Invoke()
5077 if (!can_infer_output_shape && HasUnspecifiedDimension(tensor)) { in Invoke()
5149 if (HasUnspecifiedDimension(tensor)) { in Invoke()
5186 if (!can_infer_output_shape && HasUnspecifiedDimension(tensor)) { in Invoke()