Searched defs:grad_input_ptr (Results 1 – 6 of 6) sorted by relevance
355 scalar_t* grad_input_ptr = grad_input_data + c * input_height * input_width; in cpu_padding_backward() local371 scalar_t* grad_input_ptr = grad_input_data + c * input_depth * input_height * input_width; in cpu_padding_backward() local436 scalar_t* grad_input_ptr = grad_input_data + in cpu_padding_backward_channels_last() local455 scalar_t* grad_input_ptr = grad_input_data + in cpu_padding_backward_channels_last() local
277 scalar_t* grad_input_ptr = grad_input_data + c * input_height * input_width; in cpu_adaptive_avg_pool2d_backward() local328 scalar_t* grad_input_ptr = grad_input_data + n * input_height * input_width * channels; in cpu_adaptive_avg_pool2d_backward_channels_last() local704 scalar_t* grad_input_ptr = grad_input_data + c * input_depth * input_height * input_width; in cpu_adaptive_avg_pool3d_backward() local764 …scalar_t* grad_input_ptr = grad_input_data + n * input_depth * input_height * input_width * channe… in cpu_adaptive_avg_pool3d_backward_channels_last() local
365 scalar_t* grad_input_ptr = grad_input_data + c * input_height * input_width; in cpu_adaptive_max_pool2d_backward() local413 scalar_t* grad_input_ptr = grad_input_data + n * input_height * input_width * channels; in cpu_adaptive_max_pool2d_backward_channels_last() local857 scalar_t* grad_input_ptr = grad_input_data + c * input_depth * input_height * input_width; in cpu_adaptive_max_pool3d_backward() local909 …scalar_t* grad_input_ptr = grad_input_data + n * input_depth * input_height * input_width * channe… in cpu_adaptive_max_pool3d_backward_channels_last() local
374 scalar_t* grad_input_ptr = grad_input_data + c * input_height * input_width; in cpu_avg_pool2d_backward() local443 scalar_t* grad_input_ptr = grad_input_data + n * input_height * input_width * channels; in cpu_avg_pool2d_backward_channels_last() local936 scalar_t* grad_input_ptr = grad_input_data + c * input_depth * input_height * input_width; in cpu_avg_pool3d_backward() local1014 …scalar_t* grad_input_ptr = grad_input_data + n * input_depth * input_height * input_width * channe… in cpu_avg_pool3d_backward_channels_last() local
328 scalar_t* grad_input_ptr = grad_input_data_base + offset; in _vec_softmax_backward() local439 scalar_t* grad_input_ptr = grad_input_data_base + in _vec_softmax_backward() local537 scalar_t* grad_input_ptr = grad_input_data_base + offset; in _vec_log_softmax_backward() local639 scalar_t* grad_input_ptr = grad_input_data_base + offset; in _vec_log_softmax_backward() local
512 scalar_t* grad_input_ptr = grad_input_data + c * input_depth * input_height * input_width; in cpu_max_pool_backward() local572 …scalar_t* grad_input_ptr = grad_input_data + n * input_depth * input_height * input_width * channe… in cpu_max_pool_backward_channels_last() local