Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DRNN_miopen.cpp205 struct RNNDescriptors { struct
206 RNNDescriptor rnn_desc;
207 std::vector<TensorDescriptor> x_descs;
208 std::vector<TensorDescriptor> y_descs;
209 TensorDescriptor hx_desc;
210 TensorDescriptor hy_desc;
211 TensorDescriptor cx_desc;
212 TensorDescriptor cy_desc;
214 …RNNDescriptors(const RNNParams& fn, miopenHandle_t handle, Tensor x, Tensor y, Tensor hx, Tensor c… in RNNDescriptors() function
224 std::vector<miopenTensorDescriptor_t> get_descs(const std::vector<TensorDescriptor>& descs) { in get_descs()
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DRNN.cpp497 struct RNNDescriptors { struct
498 RNNDescriptor rnn_desc;
502 std::vector<TensorDescriptor> x_descs;
503 std::vector<TensorDescriptor> y_descs;
505 RNNDataDescriptor x_descs;
506 RNNDataDescriptor y_descs;
508 TensorDescriptor hx_desc;
509 TensorDescriptor hy_desc;
510 TensorDescriptor cx_desc;
511 TensorDescriptor cy_desc;
[all …]