| /frameworks/ml/nn/common/operations/ |
| RNN.cpp | 61 const uint32_t num_units = SizeOfDimension(input_weights, 0); local
71 hiddenStateShape->dimensions = { batch_size, num_units };
75 outputShape->dimensions = { batch_size, num_units };
84 const uint32_t num_units = weights_->shape().dimensions[0]; local
96 reinterpret_cast<float*>(hidden_state_in_->buffer) + b * num_units;
98 reinterpret_cast<float*>(output_->buffer) + b * num_units;
100 reinterpret_cast<float*>(hidden_state_out_->buffer) + b * num_units;
108 for (uint32_t o = 0; o < num_units; o++) {
113 for (uint32_t o = 0; o < num_units; o++) {
121 for (uint32_t o = 0; o < num_units; o++)
[all...] |
| RNNTest.cpp | 212 uint32_t num_units() const { return units_; } function in class:android::nn::wrapper::BasicRNNOpModel
325 float* golden_start = rnn_golden_output + i * rnn.num_units();
326 float* golden_end = golden_start + rnn.num_units();
|
| SVDF.cpp | 79 const uint32_t num_units = SizeOfDimension(weights_feature, 0); local
82 NN_CHECK_EQ(SizeOfDimension(weights_time, 0), num_units); local
87 NN_CHECK_EQ(SizeOfDimension(bias, 0), num_units); local
93 stateShape->dimensions = { batch_size, memory_size * num_units };
99 outputShape->dimensions = { batch_size, num_units };
109 const int num_units = weights_feature_->shape().dimensions[0]; local
122 float* output_ptr_batch = reinterpret_cast<float*>(output_->buffer) + b * num_units;
123 const float* state_in_ptr_batch = reinterpret_cast<const float*>(state_in_->buffer) + b * (memory_size - 1) * num_units;
124 float* state_out_ptr_batch = reinterpret_cast<float*>(state_out_->buffer) + b * (memory_size - 1) * num_units;
127 for (int c = 0; c < num_units; c++)
[all...] |
| SVDFTest.cpp | 254 int num_units() const { return units_; } function in class:android::nn::wrapper::SVDFOpModel
299 const int svdf_num_units = svdf.num_units();
|
| /toolchain/binutils/binutils-2.25/gas/config/ |
| tc-xtensa.h | 448 int num_units; member in struct:__anon115824
|
| /toolchain/binutils/binutils-2.25/binutils/ |
| dwarf.c | 2249 unsigned int num_units = 0; local
[all...] |
