Searched refs:axisData (Results 1 – 2 of 2) sorted by relevance
594 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare() argument609 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) { in meanPrepare()625 int32_t current = axisData[i]; in meanPrepare()631 int32_t previous = axisData[j]; in meanPrepare()647 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,