Searched refs:numInputDims (Results 1 - 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
H A DStridedSlice.cpp42 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); local
43 for (int32_t idx = numInputDims - 1; idx >= 0; --idx) {
62 for (int i = numInputDims; i < kMaxDim; i++) {
68 beginMask = ReverseMaskBits(beginMask, numInputDims);
69 endMask = ReverseMaskBits(endMask, numInputDims);
70 shrinkAxisMask = ReverseMaskBits(shrinkAxisMask, numInputDims);
H A DReshape.cpp105 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); local
110 for (int32_t i = numInputDims - 1; i >= 0; --i) {
/frameworks/ml/nn/common/
H A DOperationsUtils.cpp574 uint32_t numInputDims = getNumberOfDimensions(input); local
575 NN_OPS_CHECK(numInputDims == 4);
580 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == numInputDims);
583 std::vector<uint32_t> outDims(numInputDims);
584 for (uint32_t i = 0; i < numInputDims; ++i) {
676 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(input)); local
683 std::vector<bool> shouldSqueeze(numInputDims, false);
688 for (int32_t idx = 0; idx < numInputDims; ++idx) {
696 int32_t current = squeezeDims[idx] < 0 ? squeezeDims[idx] + numInputDims
698 NN_OPS_CHECK(current >= 0 && current < numInputDims
725 uint32_t numInputDims = getNumberOfDimensions(input); local
758 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(input)); local
831 uint32_t numInputDims = getNumberOfDimensions(input); local
[all...]

Completed in 1198 milliseconds