Lines Matching defs:device
107 def test_Conv2d_large_workspace(self, device, dtype): argument
125 def test_ConvTranspose2d_large_output_padding(self, device, dtype): argument
142 def test_Conv2d_depthwise_naive_groups(self, device, dtype): argument
205 def test_Conv3d_depthwise_naive_groups(self, device, dtype): argument
268 def test_noncontig_conv_grad(self, device, dtype): argument
288 def test_conv_double_backward(self, device, dtype): argument
371 def test_conv1d_same_padding(self, device, dtype): argument
401 def test_conv3d_same_padding(self, device, dtype): argument
419 def test_conv1d_valid_padding(self, device, dtype): argument
427 def test_conv2d_valid_padding(self, device, dtype): argument
435 def test_conv3d_valid_padding(self, device, dtype): argument
443 def test_conv1d_same_padding_backward(self, device, dtype): argument
469 def test_conv2d_same_padding_backward(self, device, dtype): argument
496 def test_conv3d_same_padding_backward(self, device, dtype): argument
544 def test_conv1d_valid_padding_backward(self, device, dtype): argument
558 def test_conv1d_vs_scipy(self, device, dtype, mode): argument
589 def test_conv2d_vs_scipy(self, device, dtype, mode): argument
621 def test_conv3d_vs_scipy(self, device, dtype, mode): argument
656 def test_conv2d_valid_padding_backward(self, device, dtype): argument
668 def test_conv3d_valid_padding_backward(self, device, dtype): argument
691 def test_conv_transpose_with_output_size_and_no_batch_dim(self, device, N): argument
702 def test_conv_empty_channel(self, device, dtype): argument
728 def test_group_conv_empty(self, device): argument
735 def test_group_convTranspose_empty(self, device): argument
742 def test_convTranspose_empty(self, device): argument
749 def test_conv_large_nosplit(self, device): argument
758 def test_conv_noncontig_weights(self, device): argument
774 def test_conv_noncontig_weights_and_bias(self, device): argument
799 def test_conv_transposed_large(self, device): argument
833 def test_conv_large(self, device): argument
856 def test_Conv2d_size_1_kernel(self, device): argument
886 def test_ConvTranspose2d_size_1_kernel(self, device): argument
914 def test_ConvTranspose3d_size_1_kernel(self, device): argument
944 def test_Conv2d_naive_groups(self, device, dtype): argument
986 def test_Conv2d_backward_depthwise(self, device, dtype): argument
998 def test_conv_cudnn_nhwc(self, device, dtype): argument
1047 def test_conv_cudnn_ndhwc(self, device, dtype): argument
1100 device, argument
1134 def _test_conv_cudnn_nhwc_nchw(self, layer, n, c, h, w, k, filter_size, device): argument
1165 def test_conv_cudnn_nhwc_support(self, device, dtype): argument
1178 def test_conv2d_no_grad(self, device, dtype): argument
1195 def test_conv_double_backward_strided_with_3D_input_and_weight(self, device): argument
1248 def test_channels_last_ouput_stride(self, device, dtype): argument