Searched defs:inputDims (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Concatenation.cpp | 58 std::vector<tflite::Dims<4>> inputDims(num_inputs); in concatenation() local
80 std::vector<tflite::Dims<4>> inputDims(num_inputs); in concatenation() local
|
| D | MirrorPad.cpp | 193 const uint32_t inputDims = getNumberOfDimensions(inputShape); in eval() local
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3001 uint32_t inputDims[2] = {5, 5}; in randomMultinomialOpTest() local
3329 uint32_t inputDims[4] = {4, 4, 4, 4}; in instanceNormalizationOpTest() local
3534 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest() local
3969 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in bidirectionlSequenceRNNTest() local
4045 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionlSequenceRNNTest() local
4111 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionalSequenceLSTMTest() local
|
| D | TestValidation.cpp | 2546 const uint32_t inputDims[] = {kLargeDim, 1}; in TEST_F() local
|