Lines Matching full:unsqueeze

553   // template function assumes batched tensors.  unsqueeze(0) will  in slow_conv_dilated2d_cpu()
556       (is_batch ? input.contiguous(memory_format) : input.contiguous().unsqueeze(0));  in slow_conv_dilated2d_cpu()
560   Tensor output_ = (is_batch ? output : output.unsqueeze(0));  in slow_conv_dilated2d_cpu()
605   // template function assumes batched tensors.  unsqueeze(0) will  in slow_conv_dilated3d_cpu()
608       (is_batch ? input.contiguous() : input.contiguous().unsqueeze(0));  in slow_conv_dilated3d_cpu()
612   Tensor output_ = (is_batch ? output : output.unsqueeze(0));  in slow_conv_dilated3d_cpu()
654   // template function assumes batched tensors.  unsqueeze(0) will  in slow_conv_dilated2d_backward_cpu()
658                 : grad_output.contiguous().unsqueeze(0));  in slow_conv_dilated2d_backward_cpu()
660       (is_batch ? input.contiguous(memory_format) : input.contiguous().unsqueeze(0));  in slow_conv_dilated2d_backward_cpu()
670       (output_mask[0] ? (is_batch ? grad_input : grad_input.unsqueeze(0))  in slow_conv_dilated2d_backward_cpu()
710   // template function assumes batched tensors.  unsqueeze(0) will  in slow_conv_dilated3d_backward_cpu()
714                 : grad_output.contiguous().unsqueeze(0));  in slow_conv_dilated3d_backward_cpu()
716       (is_batch ? input.contiguous() : input.contiguous().unsqueeze(0));  in slow_conv_dilated3d_backward_cpu()
726       (output_mask[0] ? (is_batch ? grad_input : grad_input.unsqueeze(0))  in slow_conv_dilated3d_backward_cpu()