Searched defs:outDims (Results 1 - 1 of 1) sorted by relevance

/frameworks/ml/nn/common/
H A DOperationsUtils.cpp428 std::vector<uint32_t> outDims(targetDimsSize);
438 outDims[i] = (uint32_t)value;
443 outDims[strechDim] = (uint32_t) strechValue;
450 output->dimensions = outDims;
583 std::vector<uint32_t> outDims(numInputDims);
589 outDims[i] = beforePadding + getSizeOfDimension(input, i) + afterPadding;
592 output->dimensions = outDims;
706 std::vector<uint32_t> outDims(numInputDims - numDimsSqueezed);
709 outDims[outIdx++] = getSizeOfDimension(input, inIdx);
714 output->dimensions = outDims;
848 std::vector<uint32_t> outDims; local
[all...]

Completed in 89 milliseconds