Searched refs:SetShape (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LogicalNot.cpp | 58 NN_RET_CHECK(SetShape(input, &output)); in prepare()
|
| D | Pow.cpp | 62 return SetShape(baseShape, output); in prepare()
|
| D | Neg.cpp | 66 NN_RET_CHECK(SetShape(input, &output)); in prepare()
|
| D | Elementwise.cpp | 79 NN_RET_CHECK(SetShape(input, &output)); in prepare()
|
| D | Select.cpp | 100 NN_RET_CHECK(SetShape(input1, &output)); in prepare()
|
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 80 bool SetShape(const Shape& in, Shape* out) { in SetShape() function
347 return SetShape(input, output); in floorPrepare()
400 return SetShape(input, output); in genericActivationPrepare()
404 return SetShape(input, output); in genericNormalizationPrepare()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 139 bool SetShape(const Shape& in, Shape* out);
|