Lines Matching refs:patch_dims
21 array<ptrdiff_t, 4> patch_dims; in test_simple_patch() local
23 patch_dims[0] = 1; in test_simple_patch()
24 patch_dims[1] = 1; in test_simple_patch()
25 patch_dims[2] = 1; in test_simple_patch()
26 patch_dims[3] = 1; in test_simple_patch()
29 no_patch = tensor.extract_patches(patch_dims); in test_simple_patch()
49 patch_dims[0] = 2; in test_simple_patch()
50 patch_dims[1] = 3; in test_simple_patch()
51 patch_dims[2] = 5; in test_simple_patch()
52 patch_dims[3] = 7; in test_simple_patch()
54 single_patch = tensor.extract_patches(patch_dims); in test_simple_patch()
74 patch_dims[0] = 1; in test_simple_patch()
75 patch_dims[1] = 2; in test_simple_patch()
76 patch_dims[2] = 2; in test_simple_patch()
77 patch_dims[3] = 1; in test_simple_patch()
79 twod_patch = tensor.extract_patches(patch_dims); in test_simple_patch()
119 patch_dims[0] = 1; in test_simple_patch()
120 patch_dims[1] = 2; in test_simple_patch()
121 patch_dims[2] = 3; in test_simple_patch()
122 patch_dims[3] = 5; in test_simple_patch()
124 threed_patch = tensor.extract_patches(patch_dims); in test_simple_patch()