Home
last modified time | relevance | path

Searched defs:gig (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DRNN.cu228 accscalar_t gig = gcx * H2F(cg); in lstm_cell_backward() local
345 accscalar_t gig = H2F(go)*( H2F(hx)-H2F(ng) )*( 1-H2F(ig) )*H2F(ig); in gru_cell_backward() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DRNN.cpp1586 Tensor gig = gcx * c; in _thnn_differentiable_lstm_cell_backward() local
1628 Tensor gig = at::sigmoid_backward(grad_hy * (hx - ng), ig); in _thnn_differentiable_gru_cell_backward() local