Lines Matching refs:numInputDims
503 uint32_t numInputDims = getNumberOfDimensions(input); in padPrepare() local
508 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == numInputDims); in padPrepare()
511 std::vector<uint32_t> outDims(numInputDims); in padPrepare()
512 for (uint32_t i = 0; i < numInputDims; ++i) { in padPrepare()
598 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(input)); in meanPrepare() local
603 std::vector<uint32_t> outDims(numInputDims); in meanPrepare()
604 for (int32_t idx = 0; idx < numInputDims; ++idx) { in meanPrepare()
607 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) { in meanPrepare()
625 current += numInputDims; in meanPrepare()
627 NN_OPS_CHECK(current >= 0 && current < numInputDims); in meanPrepare()
631 previous += numInputDims; in meanPrepare()
640 std::vector<uint32_t> outDims(numInputDims - numReduceAxis); in meanPrepare()
642 for (int32_t idx = 0; idx < numInputDims; ++idx) { in meanPrepare()
645 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) { in meanPrepare()