Searched refs:gate_weight_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | gru_cell.cc | 34 const RuntimeShape& gate_weight_shape, const float* gate_weight, in GruCell() argument
63 FullyConnected(fc_params, concat_shape, concat, gate_weight_shape, in GruCell()
|
| D | gru_cell.h | 29 const RuntimeShape& gate_weight_shape, const float* gate_weight,
|
| D | unidirectional_sequence_gru.cc | 47 const RuntimeShape gate_weight_shape = GetTensorShape(gate_weight); in GruImpl() local
68 gate_weight_shape, gate_weight_data, gate_bias_shape, gate_bias_data, in GruImpl()
|