Searched defs:cx_opt (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 32 const Tensor& hx, const std::optional<Tensor>& cx_opt, in miopen_rnn()
41 …_t weight_stride0, const Tensor& weight_buf, const Tensor& hx, const std::optional<Tensor>& cx_opt, in miopen_rnn_backward()
447 const Tensor& hx, const std::optional<Tensor>& cx_opt, in miopen_rnn()
761 …_t weight_stride0, const Tensor& weight_buf, const Tensor& hx, const std::optional<Tensor>& cx_opt, in miopen_rnn_backward()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/miopen/ |
| H A D | AutocastRNN.cpp | 17 const std::optional<Tensor>& cx_opt, in miopen_rnn()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 56 const std::optional<Tensor>& cx_opt, in _cudnn_rnn()
76 const std::optional<Tensor>& cx_opt, in _cudnn_rnn_backward()
1387 const std::optional<Tensor>& cx_opt, in _cudnn_rnn()
2094 const std::optional<Tensor>& cx_opt, in _cudnn_rnn_backward()
|