Home
last modified time | relevance | path

Searched defs:bias_t_maybe_owned (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DBatchNorm_miopen.cpp65 c10::MaybeOwned<Tensor> bias_t_maybe_owned = at::borrow_from_optional_tensor(bias_t_opt); in miopen_batch_norm() local
H A DConv_miopen.cpp803 c10::MaybeOwned<Tensor> bias_t_maybe_owned = at::borrow_from_optional_tensor(bias_t_opt); in miopen_convolution() local
904 c10::MaybeOwned<Tensor> bias_t_maybe_owned = at::borrow_from_optional_tensor(bias_t_opt); in miopen_depthwise_convolution() local
1471 c10::MaybeOwned<Tensor> bias_t_maybe_owned = at::borrow_from_optional_tensor(bias_t_opt); in miopen_convolution_transpose() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DBatchNorm.cpp131 c10::MaybeOwned<Tensor> bias_t_maybe_owned = in cudnn_batch_norm() local