Lines Matching refs:outDims
393 std::vector<uint32_t> outDims(targetDimsSize); in reshapePrepare() local
403 outDims[i] = (uint32_t)value; in reshapePrepare()
408 outDims[strechDim] = (uint32_t)strechValue; in reshapePrepare()
415 output->dimensions = outDims; in reshapePrepare()
513 std::vector<uint32_t> outDims(numInputDims); in padPrepare() local
519 outDims[i] = beforePadding + getSizeOfDimension(input, i) + afterPadding; in padPrepare()
522 output->dimensions = outDims; in padPrepare()
605 std::vector<uint32_t> outDims(numInputDims); in meanPrepare() local
615 outDims[idx] = 1; in meanPrepare()
617 outDims[idx] = getSizeOfDimension(input, idx); in meanPrepare()
620 output->dimensions = outDims; in meanPrepare()
642 std::vector<uint32_t> outDims(numInputDims - numReduceAxis); in meanPrepare() local
654 outDims[idx - numSkipAxis] = getSizeOfDimension(input, idx); in meanPrepare()
658 if (outDims.empty()) { in meanPrepare()
659 outDims.push_back(1); in meanPrepare()
661 output->dimensions = outDims; in meanPrepare()