Searched refs:calculateBroadcastedShape (Results 1 – 9 of 9) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | UtilsTest.cpp | 43 EXPECT_TRUE(calculateBroadcastedShape(shape1, shape2, &actualOutputShape)); in TEST()
46 EXPECT_TRUE(calculateBroadcastedShape(shape2, shape1, &actualOutputShape)); in TEST()
57 EXPECT_FALSE(calculateBroadcastedShape(shape1, shape2, &actualOutputShape)); in TEST()
58 EXPECT_FALSE(calculateBroadcastedShape(shape2, shape1, &actualOutputShape)); in TEST()
|
| D | OperationsUtils.cpp | 341 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out) { in calculateBroadcastedShape() function
|
| /frameworks/ml/nn/common/operations/ |
| D | Pow.cpp | 67 return calculateBroadcastedShape(baseShape, exponentShape, output); in prepare()
|
| D | LogicalAndOr.cpp | 78 NN_RET_CHECK(calculateBroadcastedShape(input1, input2, &output)); in prepare()
|
| D | MaximumMinimum.cpp | 94 return calculateBroadcastedShape(in1, in2, out); in prepare()
|
| D | PRelu.cpp | 124 NN_RET_CHECK(calculateBroadcastedShape(input, alpha, &output)); in prepare()
|
| D | Comparisons.cpp | 152 NN_RET_CHECK(calculateBroadcastedShape(input1, input2, &output)); in prepare()
|
| D | Broadcast.cpp | 485 NN_RET_CHECK(calculateBroadcastedShape(input1, input2, &output)); in prepare()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 289 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out);
|