Searched refs:axisShape (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/operations/ |
| D | SimpleMath.cpp | 34 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16() argument
41 meanGeneric<float, float>(inputDataFloat32.data(), inputShape, axis, axisShape, keepDims, in meanFloat16()
48 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric() argument
55 int32_t axisSize = static_cast<int32_t>(getSizeOfDimension(axisShape, 0)); in meanGeneric()
78 const int32_t* axis, const Shape& axisShape, bool keepDims,
81 const int32_t* axis, const Shape& axisShape,
85 const int32_t* axis, const Shape& axisShape,
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | Operations.h | 114 const Shape& axisShape, bool keepDims, _Float16* outputData,
117 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape,
|
| D | OperationsUtils.h | 312 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims,
|
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsUtils.cpp | 592 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare() argument
595 NN_OPS_CHECK(axisShape.type == OperandType::TENSOR_INT32); in meanPrepare()
596 NN_OPS_CHECK(getNumberOfDimensions(axisShape) == 1); in meanPrepare()
599 int32_t axisSize = static_cast<int32_t>(getSizeOfDimension(axisShape, 0)); in meanPrepare()
|