Searched refs:CalculateShapeForBroadcast (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util_test.cc | 125 EXPECT_NE(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
136 EXPECT_NE(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
147 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
153 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
162 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
169 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
179 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
186 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
196 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
203 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
[all …]
|
| D | kernel_util.h | 306 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context,
313 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context,
|
| D | logical.cc | 80 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | floor_div.cc | 89 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | pow.cc | 81 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | floor_mod.cc | 85 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | select.cc | 112 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in SelectPrepare()
|
| D | kernel_util.cc | 456 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context, in CalculateShapeForBroadcast() function
487 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context, in CalculateShapeForBroadcast() function
|
| D | maximum_minimum.cc | 70 context, CalculateShapeForBroadcast(context, op_context.input1, in Prepare()
|
| D | div.cc | 94 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | squared_difference.cc | 138 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | mul.cc | 96 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | add.cc | 111 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | comparisons.cc | 66 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in ComparisonPrepareCommon()
|
| D | sub.cc | 237 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | activations.cc | 756 context, CalculateShapeForBroadcast(context, input, alpha, &output_size)); in PreluPrepare()
|