Searched refs:calculateBroadcastedShape (Results 1 – 9 of 9) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | UtilsTest.cpp | 51 EXPECT_TRUE(calculateBroadcastedShape(shape1, shape2, &actualOutputShape)); in TEST()
54 EXPECT_TRUE(calculateBroadcastedShape(shape2, shape1, &actualOutputShape)); in TEST()
65 EXPECT_FALSE(calculateBroadcastedShape(shape1, shape2, &actualOutputShape)); in TEST()
66 EXPECT_FALSE(calculateBroadcastedShape(shape2, shape1, &actualOutputShape)); in TEST()
|
| D | OperationsExecutionUtils.cpp | 248 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out) { in calculateBroadcastedShape() function
|
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Pow.cpp | 66 return calculateBroadcastedShape(baseShape, exponentShape, output); in prepare()
|
| D | LogicalAndOr.cpp | 61 NN_RET_CHECK(calculateBroadcastedShape(input1, input2, &output)); in prepare()
|
| D | MaximumMinimum.cpp | 92 return calculateBroadcastedShape(in1, in2, out); in prepare()
|
| D | PRelu.cpp | 105 NN_RET_CHECK(calculateBroadcastedShape(input, alpha, &output)); in prepare()
|
| D | Comparisons.cpp | 124 NN_RET_CHECK(calculateBroadcastedShape(input1, input2, &output)); in prepare()
|
| D | Broadcast.cpp | 447 NN_RET_CHECK(calculateBroadcastedShape(input1, input2, &output)); in prepare()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | OperationsExecutionUtils.h | 215 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out);
|