Searched refs:SetShape (Results 1 – 9 of 9) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Elementwise.cpp | 152 NN_RET_CHECK(SetShape(input, &output)); in prepare()
160 NN_RET_CHECK(SetShape(input, &output)); in prepareFloor()
|
| D | LogicalNot.cpp | 42 NN_RET_CHECK(SetShape(input, &output)); in prepare()
|
| D | Neg.cpp | 46 NN_RET_CHECK(SetShape(input, &output)); in prepare()
|
| D | Pow.cpp | 64 return SetShape(baseShape, output); in prepare()
|
| D | Reverse.cpp | 57 NN_RET_CHECK(SetShape(inputShape, &outputShape)); in prepare()
|
| D | Select.cpp | 80 NN_RET_CHECK(SetShape(input1, &output)); in prepare()
|
| D | StridedSlice.cpp | 159 NN_RET_CHECK(SetShape(inputShape, &outputShape)); in prepare()
|
| /packages/modules/NeuralNetworks/common/types/src/ |
| D | OperationsUtils.cpp | 41 bool SetShape(const Shape& in, Shape* out) { in SetShape() function
|
| /packages/modules/NeuralNetworks/common/types/include/ |
| D | OperationsUtils.h | 168 bool SetShape(const Shape& in, Shape* out);
|