Searched refs:meanPrepare (Results 1 – 3 of 3) sorted by relevance
/frameworks/ml/nn/common/include/ |
D | OperationsUtils.h | 328 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims,
|
/frameworks/ml/nn/common/ |
D | OperationsUtils.cpp | 594 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare() function
|
D | CpuExecutor.cpp | 1365 if (!meanPrepare(input.shape(), reinterpret_cast<const int32_t*>(axis.buffer), in executeOperation()
|