Home
last modified time | relevance | path

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

/frameworks/base/tests/Camera2Tests/SmartCamera/SimpleCamera/src/androidx/media/filterfw/
DSobelFilter.java122 int[] inputDims = inputImage.getDimensions(); in onProcess() local
125 magnitudePort.fetchAvailableFrame(inputDims).asFrameImage2D() : null; in onProcess()
127 directionPort.fetchAvailableFrame(inputDims).asFrameImage2D() : null; in onProcess()
129 FrameImage2D gxFrame = Frame.create(mImageType, inputDims).asFrameImage2D(); in onProcess()
130 FrameImage2D gyFrame = Frame.create(mImageType, inputDims).asFrameImage2D(); in onProcess()
131 mGradientXShader.setUniformValue("pix", new float[] {1f/inputDims[0], 1f/inputDims[1]}); in onProcess()
132 mGradientYShader.setUniformValue("pix", new float[] {1f/inputDims[0], 1f/inputDims[1]}); in onProcess()
/frameworks/ml/nn/common/operations/
DConcatenation.cpp54 std::vector<tflite::Dims<4>> inputDims(num_inputs); in concatenation() local
56 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenation()
57 inputDimsPtr[i] = &inputDims[i]; in concatenation()
76 std::vector<tflite::Dims<4>> inputDims(num_inputs); in concatenation() local
80 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenation()
81 inputDimsPtr[i] = &inputDims[i]; in concatenation()
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp2963 uint32_t inputDims[2] = {5, 5}; in randomMultinomialOpTest() local
2966 .dimensions = inputDims, in randomMultinomialOpTest()
3291 uint32_t inputDims[4] = {4, 4, 4, 4}; in instanceNormalizationOpTest() local
3292 ANeuralNetworksOperandType input = getOpType(inputOperandType, 4, inputDims); in instanceNormalizationOpTest()
3496 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest() local
3497 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDims); in detectionPostprocessingOpTest()
3931 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in bidirectionlSequenceRNNTest() local
3940 .dimensions = inputDims, in bidirectionlSequenceRNNTest()
4007 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionlSequenceRNNTest() local
4016 .dimensions = inputDims, in unidirectionlSequenceRNNTest()
[all …]
DTestValidation.cpp1989 const uint32_t inputDims[] = {kLargeDim, 1}; in TEST_F() local
1993 .dimensionCount = std::size(inputDims), in TEST_F()
1994 .dimensions = inputDims, in TEST_F()