Searched defs:grad_output_ptr (Results 1 – 9 of 9) sorted by relevance
306 const scalar_t* grad_output_ptr = grad_output_data_base + offset; in _vec_softmax_backward() local326 const scalar_t* grad_output_ptr = grad_output_data_base + offset; in _vec_softmax_backward() local399 const scalar_t* grad_output_ptr = grad_output_data_base + offset; in _vec_softmax_backward() local515 const scalar_t* grad_output_ptr = grad_output_data_base + in _vec_log_softmax_backward() local535 const scalar_t* grad_output_ptr = grad_output_data_base + offset; in _vec_log_softmax_backward() local604 const scalar_t* grad_output_ptr = grad_output_data_base + in _vec_log_softmax_backward() local
354 const scalar_t* grad_output_ptr = grad_output_data + c * output_height * output_width; in cpu_padding_backward() local370 …const scalar_t* grad_output_ptr = grad_output_data + c * output_depth *output_height * output_widt… in cpu_padding_backward() local438 const scalar_t* grad_output_ptr = grad_output_data + in cpu_padding_backward_channels_last() local458 const scalar_t* grad_output_ptr = grad_output_data + in cpu_padding_backward_channels_last() local
273 const scalar_t* grad_output_ptr = grad_output_data + in cpu_upsample_nearest_backward_channels_last() local306 const scalar_t* grad_output_ptr = grad_output_data + in cpu_upsample_nearest_backward_channels_last() local658 const scalar_t* grad_output_ptr = grad_output_data + in cpu_upsample_linear_backward_channels_last() local710 …const scalar_t* grad_output_ptr = grad_output_data + (n * output_depth * output_height * output_wi… in cpu_upsample_linear_backward_channels_last() local
278 const scalar_t* grad_output_ptr = grad_output_data + c * output_height * output_width; in cpu_adaptive_avg_pool2d_backward() local329 … const scalar_t* grad_output_ptr = grad_output_data + n * output_height * output_width * channels; in cpu_adaptive_avg_pool2d_backward_channels_last() local705 … scalar_t* grad_output_ptr = grad_output_data + c * output_depth * output_height * output_width; in cpu_adaptive_avg_pool3d_backward() local765 …scalar_t* grad_output_ptr = grad_output_data + n * output_depth * output_height * output_width * c… in cpu_adaptive_avg_pool3d_backward_channels_last() local
366 const scalar_t* grad_output_ptr = grad_output_data + c * output_height * output_width; in cpu_adaptive_max_pool2d_backward() local414 … const scalar_t* grad_output_ptr = grad_output_data + n * output_height * output_width * channels; in cpu_adaptive_max_pool2d_backward_channels_last() local858 … scalar_t* grad_output_ptr = grad_output_data + c * output_depth * output_height * output_width; in cpu_adaptive_max_pool3d_backward() local910 …scalar_t* grad_output_ptr = grad_output_data + n * output_depth * output_height * output_width * c… in cpu_adaptive_max_pool3d_backward_channels_last() local
375 const scalar_t* grad_output_ptr = grad_output_data + c * output_height * output_width; in cpu_avg_pool2d_backward() local444 … const scalar_t* grad_output_ptr = grad_output_data + n * output_height * output_width * channels; in cpu_avg_pool2d_backward_channels_last() local937 … scalar_t* grad_output_ptr = grad_output_data + c * output_depth * output_height * output_width; in cpu_avg_pool3d_backward() local1015 scalar_t* grad_output_ptr = grad_output_data + n * output_height * output_width * channels; in cpu_avg_pool3d_backward_channels_last() local
513 …const scalar_t* grad_output_ptr = grad_output_data + c * output_depth * output_height * output_wid… in cpu_max_pool_backward() local573 …const scalar_t* grad_output_ptr = grad_output_data + n * output_depth * output_height * output_wid… in cpu_max_pool_backward_channels_last() local
203 scalar_t* grad_output_ptr = grad_output_data + c * output_image_size; in cpu_max_unpool_backward() local
637 const scalar_t* grad_output_ptr = nullptr; // [Mq, nH, Kv] member2519 const AccessType* __restrict__ grad_output_ptr = in computeDelta() local