Home
last modified time | relevance | path

Searched defs:bias_t_opt (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DBatchNorm_miopen.cpp61 …const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, const std:… in miopen_batch_norm()
H A DConv_miopen.cpp798 const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, in miopen_convolution()
899 const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, in miopen_depthwise_convolution()
1466 const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, in miopen_convolution_transpose()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DBatchNorm.cpp124 const std::optional<Tensor>& bias_t_opt, in cudnn_batch_norm()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp1302 auto bias_t_opt = in convolution_jvp() local
1344 auto bias_t_opt = in _convolution_jvp() local