Searched refs:mkldnn_rnn_layer_backward (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | RNN.cpp | 47 std::tuple<Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor> mkldnn_rnn_layer_backward( in mkldnn_rnn_layer_backward() function
294 std::tuple<Tensor, Tensor, Tensor, Tensor, Tensor, Tensor, Tensor> mkldnn_rnn_layer_backward( in mkldnn_rnn_layer_backward() function
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_has_decomposition.expect | 981 aten::mkldnn_rnn_layer_backward
982 aten::mkldnn_rnn_layer_backward.out
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 6157 @register_meta(aten.mkldnn_rnn_layer_backward.default)
6158 def mkldnn_rnn_layer_backward( function
|