Home
last modified time | relevance | path

Searched refs:CalculateShapeForBroadcast (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util_test.cc93 EXPECT_NE(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
102 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
108 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
117 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
124 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
134 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
141 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
Dlogical.cc71 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
Dsquared_difference.cc70 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
Dfloor_div.cc77 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
Dpow.cc70 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
Dmaximum_minimum.cc63 context, CalculateShapeForBroadcast(context, op_context.input1, in Prepare()
Dkernel_util.h147 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context,
Dfloor_mod.cc95 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
Ddiv.cc71 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
Dkernel_util.cc207 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context, in CalculateShapeForBroadcast() function
Dmul.cc80 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
Dadd.cc90 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
Dcomparisons.cc51 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in ComparisonPrepare()
Dsub.cc208 TF_LITE_ENSURE_OK(context, CalculateShapeForBroadcast( in Prepare()
Dactivations.cc320 context, CalculateShapeForBroadcast(context, input, alpha, &output_size)); in PreluPrepare()