Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/
DOperationsUtils.cpp592 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare() argument
607 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) { in meanPrepare()
623 int32_t current = axisData[i]; in meanPrepare()
629 int32_t previous = axisData[j]; in meanPrepare()
645 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) { in meanPrepare()
/packages/modules/NeuralNetworks/common/include/
DOperationsUtils.h312 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims,