Home
last modified time | relevance | path

Searched defs:batch_size (Results 1 – 10 of 10) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DMultinomial.cpp78 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local
112 const uint32_t batch_size = SizeOfDimension(input_, 0); in EvalFloat32() local
DSVDF.cpp83 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local
170 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32() local
DRNN.cpp61 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local
145 const uint32_t batch_size = inputShape.dimensions[0]; in RNNStep() local
DMultinomialTest.cpp44 MultinomialOpModel(uint32_t batch_size, uint32_t class_size, uint32_t sample_size) in MultinomialOpModel()
DFullyConnected.cpp62 uint32_t batch_size = getSizeOfDimension(outputShape, 0); in fullyConnectedFloat32() local
/packages/modules/NeuralNetworks/common/types/operations/src/
DFullyConnected.cpp46 uint32_t batch_size = 0; in validateShapes() local
/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
Dqlstm_noprojection.mod.py22 batch_size = 2 variable
Dqlstm_projection.mod.py22 batch_size = 2 variable
/packages/modules/RemoteKeyProvisioning/app/src/com/android/rkpdapp/provisioner/
DProvisioner.java151 int batch_size = keysGenerated.size(); in batchProvision() local
/packages/modules/Virtualization/virtualizationservice/src/
Dmaintenance.rs51 batch_size: usize, field