Home
last modified time | relevance | path

Searched refs:outDims (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/common/
DOperationsUtils.cpp391 std::vector<uint32_t> outDims(targetDimsSize); in reshapePrepare() local
401 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() local
517 outDims[i] = beforePadding + getSizeOfDimension(input, i) + afterPadding; in padPrepare()
520 output->dimensions = outDims; in padPrepare()
603 std::vector<uint32_t> outDims(numInputDims); in meanPrepare() local
613 outDims[idx] = 1; in meanPrepare()
615 outDims[idx] = getSizeOfDimension(input, idx); in meanPrepare()
[all …]
/packages/modules/NeuralNetworks/common/operations/
DSqueeze.cpp110 std::vector<uint32_t> outDims(numInputDims - numDimsSqueezed); in prepare() local
113 outDims.push_back(1); in prepare()
117 outDims[outIdx++] = getSizeOfDimension(inputShape, inIdx); in prepare()
122 outputShape.dimensions = outDims; in prepare()
DStridedSlice.cpp172 std::vector<uint32_t> outDims; in prepare() local
189 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()
DTranspose.cpp130 std::vector<uint32_t> outDims(numInputDims); in prepare() local
133 outDims[idx] = getSizeOfDimension(input, permData[idx]); in prepare()
135 output.dimensions = outDims; in prepare()