Searched refs:outputDims (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/shim_and_sl/ |
| D | ShimPreparedModel.cpp | 351 std::vector<uint32_t> outputDims; in executeSynchronouslyInternal() local
352 auto result = execution->getOutputOperandDimensions(i, &outputDims); in executeSynchronouslyInternal()
355 outputShape.dimensions.assign(outputDims.begin(), outputDims.end()); in executeSynchronouslyInternal()
359 outputShape.dimensions.assign(outputDims.begin(), outputDims.end()); in executeSynchronouslyInternal()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3018 uint32_t outputDims[2] = {5, 7}; in randomMultinomialOpTest() local
3021 .dimensions = outputDims, in randomMultinomialOpTest()
3974 uint32_t outputDims[2] = {batchSize, numUnits}; in bidirectionlSequenceRNNTest() local
4007 .dimensions = outputDims, in bidirectionlSequenceRNNTest()
4050 uint32_t outputDims[2] = {batchSize, numUnits}; in unidirectionlSequenceRNNTest() local
4079 .dimensions = outputDims, in unidirectionlSequenceRNNTest()
4120 uint32_t outputDims[3] = {maxTime, batchSize, outputSize}; in unidirectionalSequenceLSTMTest() local
4215 .dimensions = outputDims, in unidirectionalSequenceLSTMTest()
|