Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DRNN.cu521 c10::MaybeOwned<Tensor> input_bias_maybe_owned = at::borrow_from_optional_tensor(input_bias_opt); in _thnn_fused_lstm_cell_cuda() local
607 c10::MaybeOwned<Tensor> input_bias_maybe_owned = at::borrow_from_optional_tensor(input_bias_opt); in _thnn_fused_gru_cell_cuda() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DRNN.cpp1604 c10::MaybeOwned<Tensor> input_bias_maybe_owned = at::borrow_from_optional_tensor(input_bias_opt); in _thnn_differentiable_gru_cell_backward() local