Searched refs:batch_size (Results 1 - 5 of 5) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| H A D | RNN.cpp | 60 const uint32_t batch_size = SizeOfDimension(input, 0); local
71 hiddenStateShape->dimensions = { batch_size, num_units };
75 outputShape->dimensions = { batch_size, num_units };
83 const uint32_t batch_size = input_->shape().dimensions[0]; local
91 for (uint32_t b = 0; b < batch_size; b++) {
|
| H A D | SVDF.cpp | 78 const uint32_t batch_size = SizeOfDimension(input, 0); local
93 stateShape->dimensions = { batch_size, memory_size * num_units };
99 outputShape->dimensions = { batch_size, num_units };
107 const int batch_size = input_->shape().dimensions[0]; local
119 for (int b = 0; b < batch_size; b++) {
|
| H A D | Activation.cpp | 75 uint32_t batch_size = getSizeOfDimension(inputShape, 0); local
76 uint32_t input_size = getNumberOfElements(inputShape) / batch_size;
79 shapeIn4D.dimensions = {batch_size, 1, 1, input_size};
166 uint32_t batch_size = getSizeOfDimension(inputShape, 0); local
167 uint32_t input_size = getNumberOfElements(inputShape) / batch_size;
170 shapeIn4D.dimensions = {batch_size, 1, 1, input_size};
|
| /frameworks/ml/nn/common/ |
| H A D | OperationsUtils.cpp | 334 uint32_t batch_size = input_size / getSizeOfDimension(weights, 1);
337 NN_OPS_CHECK(getSizeOfDimension(weights, 1) * batch_size == input_size);
341 output->dimensions = {batch_size, num_units};
|
| /frameworks/ml/nn/common/operations/internal/optimized/ |
| H A D | optimized_ops.h | 894 const int batch_size = ArraySize(output_dims, 3); local
899 for (int batch = 0; batch < batch_size; ++batch) {
1003 const int batch_size = ArraySize(input_dims, 3); local
1008 for (int batch = 0; batch < batch_size; ++batch) {
|
Completed in 87 milliseconds