Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DRNN.cu418 auto grad_hyI = tryGetTensorInfo<scalar_t, index_type>(grad_hy); in lstm_backward_impl() local
488 auto grad_hyI = getTensorInfo<scalar_t, index_type>(grad_hy); in gru_backward_impl() local