Searched defs:batch_size (Results 1 - 4 of 4) 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 | 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};
|
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++) {
|
/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 230 milliseconds