Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DRNN_miopen.cpp259 void _viewOrCopyParams(MatrixRef<Tensor> params_from, MatrixRef<Tensor> params_to, bool copy) { in _viewOrCopyParams()
281 void _copyParams_and_permute(MatrixRef<Tensor> params_from, MatrixRef<Tensor> params_to, int64_t mo… in _copyParams_and_permute()
297 void _copyParams(MatrixRef<Tensor> params_from, MatrixRef<Tensor> params_to) { in _copyParams()
301 void _viewParams(MatrixRef<Tensor> params_from, MatrixRef<Tensor> params_to) { in _viewParams()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DRNN.cpp1009 MatrixRef<Tensor> params_from, in get_parameters()
1053 void _copyParams(MatrixRef<Tensor> params_from, MatrixRef<Tensor> params_to) { in get_parameters()
1057 void _viewParams(MatrixRef<Tensor> params_from, MatrixRef<Tensor> params_to) { in get_parameters()