Home
last modified time | relevance | path

Searched refs:getNumberOfDimensions (Results 1 – 25 of 40) sorted by relevance

12

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp185 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
193 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
198 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
204 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2); in prepare()
210 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2); in prepare()
214 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2); in prepare()
220 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2); in prepare()
226 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2); in prepare()
230 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2); in prepare()
244 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1); in prepare()
[all …]
DBidirectionalSequenceRNN.cpp323 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3); in prepare()
324 NN_RET_CHECK_EQ(getNumberOfDimensions(fwWeights), 2); in prepare()
325 NN_RET_CHECK_EQ(getNumberOfDimensions(fwRecurrentWeights), 2); in prepare()
326 NN_RET_CHECK_EQ(getNumberOfDimensions(fwBias), 1); in prepare()
327 NN_RET_CHECK_EQ(getNumberOfDimensions(fwHiddenState), 2); in prepare()
328 NN_RET_CHECK_EQ(getNumberOfDimensions(bwWeights), 2); in prepare()
329 NN_RET_CHECK_EQ(getNumberOfDimensions(bwRecurrentWeights), 2); in prepare()
330 NN_RET_CHECK_EQ(getNumberOfDimensions(bwBias), 1); in prepare()
331 NN_RET_CHECK_EQ(getNumberOfDimensions(bwHiddenState), 2); in prepare()
348 NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3); in prepare()
[all …]
DSlice.cpp52 std::vector<uint32_t> outputIndex(getNumberOfDimensions(outputShape), 0); in evalGeneric()
54 std::vector<uint32_t> inputIndex(getNumberOfDimensions(inputShape)); in evalGeneric()
95 const int32_t n_dims = getNumberOfDimensions(inputShape); in prepare()
99 NN_RET_CHECK_EQ(getNumberOfDimensions(beginShape), 1); in prepare()
103 NN_RET_CHECK_EQ(getNumberOfDimensions(sizeShape), 1); in prepare()
DSimpleMath.cpp70 int32_t* scratchBuffer = new int32_t[getNumberOfDimensions(inputShape)]; in meanGeneric()
85 getNumberOfDimensions(inputShape), outputData, in meanGeneric()
87 getNumberOfDimensions(outputShape), axis, axisSize, keepDims, scratchBuffer, in meanGeneric()
DUnidirectionalSequenceRNN.cpp147 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3); in prepare()
148 NN_RET_CHECK_EQ(getNumberOfDimensions(weights), 2); in prepare()
149 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentWeights), 2); in prepare()
150 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in prepare()
151 NN_RET_CHECK_EQ(getNumberOfDimensions(hiddenState), 2); in prepare()
DL2Normalization.cpp45 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normFloat32Impl()
72 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normQuant8Impl()
100 int32_t ndim = getNumberOfDimensions(inputShape); in l2normFloat32()
129 int32_t ndim = getNumberOfDimensions(inputShape); in l2normQuant8()
171 int32_t numDimensions = getNumberOfDimensions(input); in prepare()
DConcatenation.cpp50 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(), in concatenation()
75 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(), in concatenation()
131 uint32_t numDimensions = getNumberOfDimensions(input0); in prepare()
139 NN_RET_CHECK_EQ(getNumberOfDimensions(input), numDimensions); in prepare()
DGather.cpp46 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
85 output.dimensions.reserve(getNumberOfDimensions(input) + getNumberOfDimensions(indices) - 1); in prepare()
DSoftmax.cpp48 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in softmaxSlowFloat32()
76 int32_t ndim = getNumberOfDimensions(inputShape); in softmaxFloat32()
122 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in softmaxQuant8Impl()
188 int32_t ndim = getNumberOfDimensions(inputShape); in softmaxQuant8()
259 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in prepare()
DNormalization.cpp37 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in localResponseNormFloat32Impl()
77 int32_t ndim = getNumberOfDimensions(inputShape); in localResponseNormFloat32()
DExpandDims.cpp28 NN_CHECK(handleNegativeAxis(getNumberOfDimensions(input) + 1, &axis)); in prepare()
DReduce.cpp50 const uint32_t inputRank = getNumberOfDimensions(inputShape); in compute()
107 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
112 NN_RET_CHECK_EQ(getNumberOfDimensions(axesShape), 1u); in prepare()
DTranspose.cpp91 uint32_t numInputDims = getNumberOfDimensions(input); in prepare()
110 NN_RET_CHECK_EQ(getNumberOfDimensions(permShape), 1); in prepare()
DArgMinMax.cpp36 inputShape, axis + 1, getNumberOfDimensions(inputShape)); in argMinMaxImpl()
DStridedSlice.cpp44 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); in stridedSliceGeneric()
DSplit.cpp37 int concatDimensions = getNumberOfDimensions(inputShape); in splitGeneric()
DGenerateProposals.cpp200 NN_RET_CHECK_EQ(getNumberOfDimensions(roiShape), 2); in prepare()
201 NN_RET_CHECK_EQ(getNumberOfDimensions(bboxDeltasShape), 2); in prepare()
202 NN_RET_CHECK_EQ(getNumberOfDimensions(batchesShape), 1); in prepare()
203 NN_RET_CHECK_EQ(getNumberOfDimensions(imageInfoShape), 2); in prepare()
681 NN_RET_CHECK(getNumberOfDimensions(scoreShape) == 2); in prepare()
682 NN_RET_CHECK(getNumberOfDimensions(roiShape) == 2); in prepare()
683 NN_RET_CHECK(getNumberOfDimensions(batchesShape) == 1); in prepare()
1174 NN_RET_CHECK_EQ(getNumberOfDimensions(scoreShape), 4); in prepare()
1175 NN_RET_CHECK_EQ(getNumberOfDimensions(bboxDeltasShape), 4); in prepare()
1176 NN_RET_CHECK_EQ(getNumberOfDimensions(anchorsShape), 2); in prepare()
[all …]
DFullyConnected.cpp208 NN_RET_CHECK_GE(getNumberOfDimensions(input), 2); in prepare()
209 NN_RET_CHECK_EQ(getNumberOfDimensions(weights), 2); in prepare()
DTransposeConv2D.cpp68 NN_RET_CHECK_EQ(getNumberOfDimensions(context->getInputShape(3)), 1); in initialize()
480 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 4); in prepare()
481 NN_RET_CHECK_EQ(getNumberOfDimensions(filter), 4); in prepare()
482 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in prepare()
DChannelShuffle.cpp44 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
/frameworks/ml/nn/common/
DOperationsUtils.cpp136 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions() function
314 uint32_t numberOfDims1 = getNumberOfDimensions(in1); in calculateBroadcastedShape()
315 uint32_t numberOfDims2 = getNumberOfDimensions(in2); in calculateBroadcastedShape()
365 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in depthwiseConvPrepare()
366 NN_OPS_CHECK(getNumberOfDimensions(filter) == 4); in depthwiseConvPrepare()
367 NN_OPS_CHECK(getNumberOfDimensions(bias) == 1); in depthwiseConvPrepare()
399 NN_OPS_CHECK(getNumberOfDimensions(input) <= 4); in genericActivationPrepare()
449 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in depthToSpacePrepare()
472 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToDepthPrepare()
497 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 2); in embeddingLookupPrepare()
[all …]
/frameworks/ml/nn/common/include/
DCpuOperationUtils.h101 NN_RET_CHECK_EQ(getNumberOfDimensions(nchwShape), 4) in convertNchwToNhwc()
122 NN_RET_CHECK_EQ(getNumberOfDimensions(nhwcShape), 4) in convertNhwcToNchw()
168 NN_RET_CHECK_EQ(getNumberOfDimensions(shape), 4); in initialize()
DOperationsUtils.h152 uint32_t getNumberOfDimensions(const Shape& shape);
160 return handleNegativeAxis(getNumberOfDimensions(shape), axis); in handleNegativeAxis()
376 const int numDims = getNumberOfDimensions(shape); in transposeFirstTwoDimensions()
397 NN_RET_CHECK(getNumberOfDimensions(shape) >= 2); in transposeFirstTwoDimensions()
/frameworks/base/media/mca/filterfw/java/android/filterfw/core/
DNativeFrame.java170 if (getFormat().getNumberOfDimensions() != 2) { in setBitmap()
187 if (getFormat().getNumberOfDimensions() != 2) { in getBitmap()
/frameworks/base/tests/Camera2Tests/SmartCamera/SimpleCamera/src/androidx/media/filterfw/
DInputPort.java61 if(port.mInfo.type.getNumberOfDimensions() > 0) { in onFrameReceived()

12