Searched defs:grad_output_reshaped (Results 1 – 2 of 2) sorted by relevance
117 auto grad_output_reshaped = grad_output.dim() > 2 ? in mkldnn_linear_backward_input() local148 auto grad_output_reshaped = grad_output.dim() > 2 ? in mkldnn_linear_backward_weights() local
613 auto grad_output_reshaped = grad_out_.reshape(input_reshaped.sizes()); in batch_norm_backward_cuda_template() local789 auto grad_output_reshaped = grad_out_.reshape(input_reshaped.sizes()); in batch_norm_backward_reduce_cuda_template() local845 auto grad_output_reshaped = grad_out_.reshape(input_reshaped.sizes()); in batch_norm_backward_elemt_cuda_template() local896 auto grad_output_reshaped = grad_out_.reshape(input_reshaped.sizes()); in batch_norm_backward_elemt_cuda_template() local