Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DOperationsUtils.cpp56 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions() function
168 NN_OPS_CHECK(getNumberOfDimensions(in1) <= 4 && getNumberOfDimensions(in2) <= 4); in addMulPrepare()
174 uint32_t numberOfDims1 = getNumberOfDimensions(in1); in addMulPrepare()
175 uint32_t numberOfDims2 = getNumberOfDimensions(in2); in addMulPrepare()
228 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in convPrepare()
229 NN_OPS_CHECK(getNumberOfDimensions(filter) == 4); in convPrepare()
230 NN_OPS_CHECK(getNumberOfDimensions(bias) == 1); in convPrepare()
265 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in depthwiseConvPrepare()
266 NN_OPS_CHECK(getNumberOfDimensions(filter) == 4); in depthwiseConvPrepare()
267 NN_OPS_CHECK(getNumberOfDimensions(bias) == 1); in depthwiseConvPrepare()
[all …]
/frameworks/ml/nn/common/operations/
DActivation.cpp74 if (getNumberOfDimensions(inputShape) == 2) { in softmaxFloat32()
81 } else if (getNumberOfDimensions(inputShape) == 4) { in softmaxFloat32()
165 if (getNumberOfDimensions(inputShape) == 2) { in softmaxQuant8()
172 } else if (getNumberOfDimensions(inputShape) == 4) { in softmaxQuant8()
DConcatenation.cpp37 getNumberOfDimensions(outputShape) - axis - 1, in concatenationFloat32()
56 getNumberOfDimensions(outputShape) - axis - 1, in concatenationQuant8()
/frameworks/base/media/mca/filterfw/java/android/filterfw/core/
DNativeFrame.java170 if (getFormat().getNumberOfDimensions() != 2) { in setBitmap()
187 if (getFormat().getNumberOfDimensions() != 2) { in getBitmap()
DFrameFormat.java130 public int getNumberOfDimensions() { in getNumberOfDimensions() method in FrameFormat
/frameworks/base/tests/Camera2Tests/SmartCamera/SimpleCamera/src/androidx/media/filterfw/
DInputPort.java61 if(port.mInfo.type.getNumberOfDimensions() > 0) { in onFrameReceived()
DFrameType.java264 public int getNumberOfDimensions() { in getNumberOfDimensions() method in FrameType
/frameworks/ml/nn/common/include/
DOperationsUtils.h66 uint32_t getNumberOfDimensions(const Shape& shape);