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

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dbidirectional_sequence_rnn.cc70 const int bw_num_units = bw_input_weights->dims->data[0]; local
116 bw_output_size_array->data[2] = bw_num_units;
156 const int bw_num_units = bw_input_weights->dims->data[0]; local
178 bw_hidden_state->data.f + b * bw_num_units;
183 bw_output->data.f + b * bw_num_units * max_time + s * bw_num_units;
187 bw_bias_ptr, input_size, bw_num_units, /*batch_size=*/1,

Completed in 121 milliseconds