Searched defs:inputDims (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Concatenation.cpp | 54 std::vector<tflite::Dims<4>> inputDims(num_inputs); in concatenation() local
76 std::vector<tflite::Dims<4>> inputDims(num_inputs); in concatenation() local
|
| /frameworks/base/tests/Camera2Tests/SmartCamera/SimpleCamera/src/androidx/media/filterfw/ |
| D | SobelFilter.java | 122 int[] inputDims = inputImage.getDimensions(); in onProcess() local
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 2963 uint32_t inputDims[2] = {5, 5}; in randomMultinomialOpTest() local
3291 uint32_t inputDims[4] = {4, 4, 4, 4}; in instanceNormalizationOpTest() local
3496 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest() local
3931 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in bidirectionlSequenceRNNTest() local
4007 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionlSequenceRNNTest() local
4073 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionalSequenceLSTMTest() local
|
| D | TestValidation.cpp | 1989 const uint32_t inputDims[] = {kLargeDim, 1}; in TEST_F() local
|