Home
last modified time | relevance | path

Searched refs:axisData (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/
DOperationsExecutionUtils.cpp501 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare() argument
516 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) { in meanPrepare()
532 int32_t current = axisData[i]; in meanPrepare()
538 int32_t previous = axisData[j]; in meanPrepare()
554 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) { in meanPrepare()
/packages/modules/NeuralNetworks/common/include/
DOperationsExecutionUtils.h254 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims,