Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/
DLossNLL2d.cpp317 const scalar_t grad_output_value = grad_output_acc[b][h][w]; in nll_loss2d_backward_out_frame() local
334 const scalar_t grad_output_value = *grad_output.const_data_ptr<scalar_t>(); in nll_loss2d_backward_out_frame() local
DLossNLL.cpp378 const scalar_t grad_output_value = *grad_output.const_data_ptr<scalar_t>(); in nll_loss_backward_out_frame() local
/external/pytorch/aten/src/ATen/native/cpu/
DUpSampleMoreKernel.cpp473 opmath_t grad_output_value = grad_output_data[c * output_slice_size + ow]; in cpu_upsample_linear_backward() local
511 … opmath_t grad_output_value = grad_output_data[c * output_slice_size + oh * output_width + ow]; in cpu_upsample_linear_backward() local
557 opmath_t grad_output_value = grad_output_data[c * output_slice_size + in cpu_upsample_linear_backward() local
DUpSampleKernel.cpp2006 scalar_t grad_output_value = in cpu_upsample_genNd_backward_aa() local