Home
last modified time | relevance | path

Searched refs:HaveSameShapes (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util_test.cc68 EXPECT_TRUE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
71 EXPECT_FALSE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
74 EXPECT_FALSE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
77 EXPECT_FALSE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
80 EXPECT_TRUE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
83 EXPECT_FALSE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
86 EXPECT_TRUE(HaveSameShapes(&tensor1_, &tensor2_)); in TEST_F()
Dcomparisons.cc47 bool requires_broadcast = !HaveSameShapes(input1, input2); in ComparisonPrepare()
130 bool requires_broadcast = !HaveSameShapes(input1, input2); in EqualEval()
166 bool requires_broadcast = !HaveSameShapes(input1, input2); in NotEqualEval()
201 bool requires_broadcast = !HaveSameShapes(input1, input2); in GreaterEval()
233 bool requires_broadcast = !HaveSameShapes(input1, input2); in GreaterEqualEval()
265 bool requires_broadcast = !HaveSameShapes(input1, input2); in LessEval()
297 bool requires_broadcast = !HaveSameShapes(input1, input2); in LessEqualEval()
Dselect.cc47 TF_LITE_ENSURE(context, HaveSameShapes(input_x, input_y)); in SelectPrepare()
52 bool same_shape = HaveSameShapes(input_condition, input_x); in SelectPrepare()
71 bool is_rank_one = !HaveSameShapes(input_condition, input_x); in SelectEval()
Dadd_n.cc41 TF_LITE_ENSURE(context, HaveSameShapes(input1, input)); in Prepare()
Dlogical.cc67 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
Dsquared_difference.cc66 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
Dfloor_div.cc73 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
Dpow.cc66 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
Dmaximum_minimum.cc58 !HaveSameShapes(op_context.input1, op_context.input2); in Prepare()
Dkernel_util.h143 bool HaveSameShapes(const TfLiteTensor* input1, const TfLiteTensor* input2);
Dpack.cc51 TF_LITE_ENSURE(context, HaveSameShapes(input0, input)); in Prepare()
Dfloor_mod.cc91 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
Ddiv.cc67 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
Dkernel_util.cc200 bool HaveSameShapes(const TfLiteTensor* input1, const TfLiteTensor* input2) { in HaveSameShapes() function
Dmul.cc76 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
Dadd.cc86 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
Dsub.cc204 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
Dactivations.cc326 TF_LITE_ENSURE(context, HaveSameShapes(input, output)); in PreluPrepare()