Searched refs: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
60 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenation()
61 inputDimsPtr[i] = &inputDims[i]; in concatenation()
80 std::vector<tflite::Dims<4>> inputDims(num_inputs); in concatenation() local
84 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenation()
85 inputDimsPtr[i] = &inputDims[i]; in concatenation()
|
| D | MirrorPad.cpp | 193 const uint32_t inputDims = getNumberOfDimensions(inputShape); in eval() local
194 NN_RET_CHECK(checkAsInt(inputDims)); in eval()
195 const int numDims = int(inputDims); in eval()
206 std::vector<int> outputDimsNumElements(inputDims, 1); in eval()
207 std::vector<int> inputDimsNumElements(inputDims, 1); in eval()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3001 uint32_t inputDims[2] = {5, 5}; in randomMultinomialOpTest() local
3004 .dimensions = inputDims, in randomMultinomialOpTest()
3329 uint32_t inputDims[4] = {4, 4, 4, 4}; in instanceNormalizationOpTest() local
3330 ANeuralNetworksOperandType input = getOpType(inputOperandType, 4, inputDims); in instanceNormalizationOpTest()
3534 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest() local
3535 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDims); in detectionPostprocessingOpTest()
3969 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in bidirectionlSequenceRNNTest() local
3978 .dimensions = inputDims, in bidirectionlSequenceRNNTest()
4045 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionlSequenceRNNTest() local
4054 .dimensions = inputDims, in unidirectionlSequenceRNNTest()
[all …]
|
| D | TestValidation.cpp | 2546 const uint32_t inputDims[] = {kLargeDim, 1}; in TEST_F() local
2550 .dimensionCount = std::size(inputDims), in TEST_F()
2551 .dimensions = inputDims, in TEST_F()
|