Lines Matching refs:numInputDims
412 uint32_t numInputDims = getNumberOfDimensions(input); in padPrepare() local
417 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == numInputDims); in padPrepare()
420 std::vector<uint32_t> outDims(numInputDims); in padPrepare()
421 for (uint32_t i = 0; i < numInputDims; ++i) { in padPrepare()
507 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(input)); in meanPrepare() local
512 std::vector<uint32_t> outDims(numInputDims); in meanPrepare()
513 for (int32_t idx = 0; idx < numInputDims; ++idx) { in meanPrepare()
516 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) { in meanPrepare()
534 current += numInputDims; in meanPrepare()
536 NN_OPS_CHECK(current >= 0 && current < numInputDims); in meanPrepare()
540 previous += numInputDims; in meanPrepare()
549 std::vector<uint32_t> outDims(numInputDims - numReduceAxis); in meanPrepare()
551 for (int32_t idx = 0; idx < numInputDims; ++idx) { in meanPrepare()
554 if (axisData[axisIdx] == idx || axisData[axisIdx] + numInputDims == idx) { in meanPrepare()