Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dbatch_norm_kernel.cpp424 scalar_t* grad_bias_data = grad_bias.defined() ? grad_bias.data_ptr<scalar_t>() : nullptr; in batch_norm_cpu_backward_contiguous_impl() local
545 scalar_t* grad_bias_data = grad_bias.defined() ? grad_bias.data_ptr<scalar_t>() : nullptr; in batch_norm_cpu_backward_channels_last_impl() local
1014 param_t* grad_bias_data = grad_bias.defined() ? grad_bias.data_ptr<param_t>() : nullptr; in batch_norm_cpu_backward_contiguous_internal() local
1136 param_t* grad_bias_data = grad_bias.defined() ? grad_bias.data_ptr<param_t>() : nullptr; in batch_norm_cpu_backward_channels_last_internal() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_convolution_backward.cpp132 CTYPE* grad_bias_data = nullptr; in conv2d_backward_impl() local