Searched refs:HaveSameShapes (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | comparisons.cc | 62 bool requires_broadcast = !HaveSameShapes(input1, input2); in ComparisonPrepareCommon()
172 bool requires_broadcast = !HaveSameShapes(input1, input2); in EqualEval()
222 bool requires_broadcast = !HaveSameShapes(input1, input2); in NotEqualEval()
272 bool requires_broadcast = !HaveSameShapes(input1, input2); in GreaterEval()
313 bool requires_broadcast = !HaveSameShapes(input1, input2); in GreaterEqualEval()
354 bool requires_broadcast = !HaveSameShapes(input1, input2); in LessEval()
395 bool requires_broadcast = !HaveSameShapes(input1, input2); in LessEqualEval()
|
| D | select.cc | 91 bool same_shape = HaveSameShapes(input_condition, input_x) && in SelectPrepare()
92 HaveSameShapes(input_x, input_y); in SelectPrepare()
108 TF_LITE_ENSURE(context, HaveSameShapes(input_x, input_y)); in SelectPrepare()
|
| D | kernel_util_test.cc | 100 EXPECT_TRUE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
103 EXPECT_FALSE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
106 EXPECT_FALSE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
109 EXPECT_FALSE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
112 EXPECT_TRUE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
115 EXPECT_FALSE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
118 EXPECT_TRUE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
|
| D | hashtable_import.cc | 54 TF_LITE_ENSURE(context, HaveSameShapes(key_tensor, value_tensor)); in PrepareHashtableImport()
|
| D | logical.cc | 76 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
|
| D | floor_div.cc | 85 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
|
| D | pow.cc | 77 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
|
| D | add_n.cc | 92 TF_LITE_ENSURE(context, HaveSameShapes(input1, input)); in Prepare()
|
| D | floor_mod.cc | 81 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
|
| D | pack.cc | 61 TF_LITE_ENSURE(context, HaveSameShapes(input0, input)); in Prepare()
|
| D | kernel_util.h | 292 bool HaveSameShapes(const TfLiteTensor* input1, const TfLiteTensor* input2);
|
| D | maximum_minimum.cc | 65 !HaveSameShapes(op_context.input1, op_context.input2); in Prepare()
|
| D | div.cc | 90 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
|
| D | squared_difference.cc | 134 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
|
| D | mul.cc | 92 const bool requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
|
| D | add.cc | 107 const bool requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
|
| D | sub.cc | 233 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
|
| D | kernel_util.cc | 409 bool HaveSameShapes(const TfLiteTensor* input1, const TfLiteTensor* input2) { in HaveSameShapes() function
|
| D | activations.cc | 751 data->requires_broadcast = !HaveSameShapes(input, alpha); in PreluPrepare()
762 TF_LITE_ENSURE(context, HaveSameShapes(input, output)); in PreluPrepare()
|