Searched refs:calculateBroadcastedShape (Results 1 – 9 of 9) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | UtilsTest.cpp | 39 EXPECT_TRUE(calculateBroadcastedShape(shape1, shape2, &actualOutputShape)); in TEST()
42 EXPECT_TRUE(calculateBroadcastedShape(shape2, shape1, &actualOutputShape)); in TEST()
53 EXPECT_FALSE(calculateBroadcastedShape(shape1, shape2, &actualOutputShape)); in TEST()
54 EXPECT_FALSE(calculateBroadcastedShape(shape2, shape1, &actualOutputShape)); in TEST()
|
| D | OperationsUtils.cpp | 312 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out) { in calculateBroadcastedShape() function
|
| /frameworks/ml/nn/common/operations/ |
| D | Pow.cpp | 64 return calculateBroadcastedShape(baseShape, exponentShape, output); in prepare()
|
| D | LogicalAndOr.cpp | 76 NN_RET_CHECK(calculateBroadcastedShape(input1, input2, &output)); in prepare()
|
| D | MaximumMinimum.cpp | 88 return calculateBroadcastedShape(in1, in2, out); in prepare()
|
| D | PRelu.cpp | 113 NN_RET_CHECK(calculateBroadcastedShape(input, alpha, &output)); in prepare()
|
| D | Comparisons.cpp | 141 NN_RET_CHECK(calculateBroadcastedShape(input1, input2, &output)); in prepare()
|
| D | Broadcast.cpp | 388 NN_RET_CHECK(calculateBroadcastedShape(input1, input2, &output)); in prepare()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 288 bool calculateBroadcastedShape(const Shape& in1, const Shape& in2, Shape* out);
|