Searched refs:outputDims (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/shim_and_sl/ |
| D | ShimPreparedModel.cpp | 294 std::vector<uint32_t> outputDims; in executeSynchronously() local
295 auto result = execution->getOutputOperandDimensions(i, &outputDims); in executeSynchronously()
298 outputShape.dimensions.assign(outputDims.begin(), outputDims.end()); in executeSynchronously()
302 outputShape.dimensions.assign(outputDims.begin(), outputDims.end()); in executeSynchronously()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 2979 uint32_t outputDims[2] = {5, 7}; in randomMultinomialOpTest() local
2982 .dimensions = outputDims, in randomMultinomialOpTest()
3935 uint32_t outputDims[2] = {batchSize, numUnits}; in bidirectionlSequenceRNNTest() local
3968 .dimensions = outputDims, in bidirectionlSequenceRNNTest()
4011 uint32_t outputDims[2] = {batchSize, numUnits}; in unidirectionlSequenceRNNTest() local
4040 .dimensions = outputDims, in unidirectionlSequenceRNNTest()
4081 uint32_t outputDims[3] = {maxTime, batchSize, outputSize}; in unidirectionalSequenceLSTMTest() local
4176 .dimensions = outputDims, in unidirectionalSequenceLSTMTest()
|