Searched defs:bias_g (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Normalization.cu | 725 …ensor& invstd, const std::optional<Tensor>& weight_opt, bool input_g, bool weight_g, bool bias_g) { in batch_norm_backward_reduce_cuda()
|
| D | Normalization.cuh | 780 … const bool input_g, const bool weight_g, const bool bias_g) { in batch_norm_backward_reduce_cuda_template()
1533 const bool input_g, const bool weight_g, const bool bias_g) { in batch_norm_backward_reduce_cuda_channels_last_template()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Convolution.cpp | 1658 auto bias_g = subtensor(bias, 0, params.groups, g); in _convolution() local
|