Searched defs:gru_cell (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gru_cell.cc | 25 namespace gru_cell { namespace
|
| /external/pytorch/benchmarks/fastrnns/ |
| D | cells.py | 118 def gru_cell(input, hidden, w_ih, w_hh, b_ih, b_hh): function
|
| /external/pytorch/aten/src/ATen/native/ |
| D | RNN.cpp | 1640 Tensor gru_cell( in gru_cell() function
|
| /external/pytorch/torch/_decomp/ |
| D | decompositions.py | 3525 def gru_cell(inp, cur_hidden, ih_weight, ih_bias, hh_weight, hh_bias): function
|