Searched refs:CalculateShapeForBroadcast (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util_test.cc | 123 EXPECT_NE(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
134 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
140 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
149 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
156 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
166 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
173 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
185 CalculateShapeForBroadcast(&context_, &tensor1_, &tensor2_, in TEST_F()
198 CalculateShapeForBroadcast(&context_, &tensor1_, &tensor2_, in TEST_F()
206 CalculateShapeForBroadcast(&context_, &tensor1_, &tensor2_, in TEST_F()
[all …]
|
| D | kernel_util.h | 275 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context,
282 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context,
|
| D | logical.cc | 80 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | pow.cc | 81 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | floor_div.cc | 89 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | floor_mod.cc | 85 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | select.cc | 103 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in SelectPrepare()
|
| D | kernel_util.cc | 405 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context, in CalculateShapeForBroadcast() function
434 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 | 136 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|
| D | mul.cc | 94 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 | 731 context, CalculateShapeForBroadcast(context, input, alpha, &output_size)); in PreluPrepare()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | div.cc | 94 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
|