Searched refs:gate_bias_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | gru_cell.cc | 35 const RuntimeShape& gate_bias_shape, const float* gate_bias, in GruCell() argument
64 gate_weight, gate_bias_shape, gate_bias, activation_shape, in GruCell()
|
| D | gru_cell.h | 30 const RuntimeShape& gate_bias_shape, const float* gate_bias,
|
| D | unidirectional_sequence_gru.cc | 49 const RuntimeShape gate_bias_shape = GetTensorShape(gate_bias); in GruImpl() local
68 gate_weight_shape, gate_weight_data, gate_bias_shape, gate_bias_data, in GruImpl()
|