Searched refs:outDims (Results 1 – 4 of 4) sorted by relevance
391 std::vector<uint32_t> outDims(targetDimsSize); in reshapePrepare() local401 outDims[i] = (uint32_t)value; in reshapePrepare()406 outDims[strechDim] = (uint32_t)strechValue; in reshapePrepare()413 output->dimensions = outDims; in reshapePrepare()511 std::vector<uint32_t> outDims(numInputDims); in padPrepare() local517 outDims[i] = beforePadding + getSizeOfDimension(input, i) + afterPadding; in padPrepare()520 output->dimensions = outDims; in padPrepare()603 std::vector<uint32_t> outDims(numInputDims); in meanPrepare() local613 outDims[idx] = 1; in meanPrepare()615 outDims[idx] = getSizeOfDimension(input, idx); in meanPrepare()[all …]
110 std::vector<uint32_t> outDims(numInputDims - numDimsSqueezed); in prepare() local113 outDims.push_back(1); in prepare()117 outDims[outIdx++] = getSizeOfDimension(inputShape, inIdx); in prepare()122 outputShape.dimensions = outDims; in prepare()
172 std::vector<uint32_t> outDims; in prepare() local189 outDims.push_back(outDim); in prepare()198 if (outDims.empty()) { in prepare()199 outDims.push_back(1); in prepare()204 outputShape.dimensions = outDims; in prepare()
130 std::vector<uint32_t> outDims(numInputDims); in prepare() local133 outDims[idx] = getSizeOfDimension(input, permData[idx]); in prepare()135 output.dimensions = outDims; in prepare()