Searched refs:axisData (Results 1 – 2 of 2) sorted by relevance
598 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare() argument613 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) { in meanPrepare()629 int32_t current = axisData[i]; in meanPrepare()635 int32_t previous = axisData[j]; in meanPrepare()651 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) { in meanPrepare()
328 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims,