Searched defs:output_ptr_batch (Results 1 - 2 of 2) sorted by relevance

/frameworks/ml/nn/common/operations/
H A DRNN.cpp97 float* output_ptr_batch = local
109 output_ptr_batch[o] = bias_ptr[o];
115 output_ptr_batch[o] += input_ptr_batch[i] * input_weights_ptr[i];
123 output_ptr_batch[o] +=
131 output_ptr_batch[o] =
132 (ActivationFunctor(activation_))(output_ptr_batch[o]);
133 hidden_state_out_ptr_batch[o] = output_ptr_batch[o];
H A DSVDF.cpp157 float* output_ptr_batch = GetBuffer<float>(output_) + b * num_units; local
160 scratch_ptr_batch, output_ptr_batch, num_units, rank);
165 float* output_ptr_batch = GetBuffer<float>(output_) + b * num_units; local
167 output_ptr_batch, num_units,
168 params_.activation_, output_ptr_batch);

Completed in 185 milliseconds