Home
last modified time | relevance | path

Searched defs:inputDims (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DConcatenation.cpp43 std::vector<tflite::Dims<4> > inputDims(num_inputs); in concatenation() local
65 std::vector<tflite::Dims<4> > inputDims(num_inputs); in concatenation() local
/frameworks/base/tests/Camera2Tests/SmartCamera/SimpleCamera/src/androidx/media/filterfw/
DSobelFilter.java122 int[] inputDims = inputImage.getDimensions(); in onProcess() local
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp2041 uint32_t inputDims[2] = {5, 5}; in randomMultinomialOpTest() local
2349 uint32_t inputDims[4] = {4, 4, 4, 4}; in instanceNormalizationOpTest() local
2528 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest() local
2932 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in bidirectionlSequenceRNNTest() local
3008 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionlSequenceRNNTest() local
3074 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionalSequenceLSTMTest() local