Searched defs:axisShape (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | SimpleMath.cpp | 40 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
54 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric()
|
| D | Reverse.cpp | 49 const Shape axisShape = context->getInputShape(kInputAxisTensor); in prepare() local
|
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsExecutionUtils.cpp | 501 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare()
|