Searched defs:pack_hidden (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 805 hidden_type pack_hidden(const Tensor& hx, const Tensor& cx) { in pack_hidden() function
811 Tensor pack_hidden<Tensor>(const Tensor& hx, const Tensor& cx) { in pack_hidden() function
817 std::tuple<Tensor, Tensor> pack_hidden<std::tuple<Tensor, Tensor>>(const Tensor& hx, const Tensor& … in pack_hidden() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | RNN.cpp | 525 hidden_type pack_hidden(const Tensor& hx, const Tensor& cx) { in pack_hidden() function
530 std::tuple<Tensor, Tensor> pack_hidden<std::tuple<Tensor, Tensor>>(const Tensor& hx, const Tensor& … in pack_hidden() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 2236 hidden_type pack_hidden(const Tensor& hx, const Tensor& cx) { in pack_hidden() function
2243 Tensor pack_hidden<Tensor>(const Tensor& hx, const Tensor& cx) { in pack_hidden() function
2249 std::tuple<Tensor, Tensor> pack_hidden<std::tuple<Tensor, Tensor>>( in pack_hidden() function
|