Searched defs:grad_output_contig (Results 1 – 4 of 4) sorted by relevance
390 Tensor grad_output_contig = grad_output->contiguous(memory_format); in cudnn_convolution_backward_input() local517 TensorArg grad_output_contig{grad_output_contig_t, "grad_output", 1}; in cudnn_convolution_backward_weight() local
140 auto grad_output_contig = contiguousIfZeroInStrides(grad_output_t); in cudnn_grid_sampler_backward() local
292 auto grad_output_contig = in cudnn_batch_norm_backward() local
1079 TensorArg grad_output_contig{ grad_output_contig_t, "grad_output", 1 }; in miopen_depthwise_convolution_backward_weight() local1132 TensorArg grad_output_contig{ grad_output_contig_t, "grad_output", 1 }; in miopen_convolution_backward_weight() local1294 Tensor grad_output_contig = grad_output->contiguous(memory_format); in miopen_convolution_backward_input() local1400 Tensor grad_output_contig = grad_output->contiguous(memory_format); in miopen_depthwise_convolution_backward_input() local