Searched defs:axisShape (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/operations/ |
| D | SimpleMath.cpp | 34 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
48 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric()
|
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsUtils.cpp | 592 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare()
|