Searched defs:batch_size (Results 1 – 11 of 11) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Multinomial.cpp | 78 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local
112 const uint32_t batch_size = SizeOfDimension(input_, 0); in EvalFloat32() local
|
| D | SVDF.cpp | 83 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local
170 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32() local
|
| D | RNN.cpp | 61 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local
145 const uint32_t batch_size = inputShape.dimensions[0]; in RNNStep() local
|
| D | MultinomialTest.cpp | 44 MultinomialOpModel(uint32_t batch_size, uint32_t class_size, uint32_t sample_size) in MultinomialOpModel()
|
| D | FullyConnected.cpp | 62 uint32_t batch_size = getSizeOfDimension(outputShape, 0); in fullyConnectedFloat32() local
|
| /packages/modules/NeuralNetworks/common/types/operations/src/ |
| D | FullyConnected.cpp | 46 uint32_t batch_size = 0; in validateShapes() local
|
| /packages/modules/Bluetooth/system/blueberry/tests/gd/l2cap/classic/ |
| D | l2cap_performance_test.py | 65 def _basic_mode_tx_fixed_interval(self, mtu, interval=timedelta(seconds=10), batch_size=20): argument
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | qlstm_noprojection.mod.py | 22 batch_size = 2 variable
|
| D | qlstm_projection.mod.py | 22 batch_size = 2 variable
|
| /packages/modules/RemoteKeyProvisioning/app/src/com/android/rkpdapp/provisioner/ |
| D | Provisioner.java | 147 int batch_size = keysGenerated.size(); in batchProvision() local
|
| /packages/modules/OnDevicePersonalization/federatedcompute/proto/ |
| D | plan.proto | 781 int32 batch_size = 3; field
798 string batch_size = 4; field
|