Home
last modified time | relevance | path

Searched defs:outputDims (Results 1 – 2 of 2) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/shim_and_sl/
H A DShimPreparedModel.cpp294 std::vector<uint32_t> outputDims; in executeSynchronously() local
/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestValidateOperations.cpp2979 uint32_t outputDims[2] = {5, 7}; in randomMultinomialOpTest() local
3935 uint32_t outputDims[2] = {batchSize, numUnits}; in bidirectionlSequenceRNNTest() local
4011 uint32_t outputDims[2] = {batchSize, numUnits}; in unidirectionlSequenceRNNTest() local
4081 uint32_t outputDims[3] = {maxTime, batchSize, outputSize}; in unidirectionalSequenceLSTMTest() local