Searched defs:patch_cols (Results 1 – 8 of 8) sorted by relevance
36 const int patch_cols = 2; in TEST() local85 const int patch_cols = 2; in TEST() local137 const int patch_cols = 4; in TEST() local185 const int patch_cols = 4; in TEST() local237 const int patch_cols = 5; in TEST() local292 const int patch_cols = 5; in TEST() local344 const int patch_cols = 3; in test_batched_strided_spatial_convolution_backward_input_valid() local420 const int patch_cols = 3; in test_batched_strided_spatial_convolution_backward_input_valid_row_major() local500 const int patch_cols = 2; in TEST() local554 const int patch_cols = 2; in TEST() local[all …]
33 const int patch_cols = 4; in TEST() local81 const int patch_cols = 4; in TEST() local131 const int patch_cols = 3; in TEST() local191 const int patch_cols = 3; in TEST() local251 const int patch_cols = 3; in TEST() local313 const int patch_cols = 3; in TEST() local375 const int patch_cols = 3; in TEST() local453 const int patch_cols = 3; in TEST() local530 const int patch_cols = 3; in TEST() local578 const int patch_cols = 3; in TEST() local[all …]
36 const int patch_cols = 2; in TEST() local99 const int patch_cols = 2; in TEST() local162 const int patch_cols = 2; in TEST() local233 const int patch_cols = 3; in TEST() local305 const int patch_cols = 2; in TEST() local372 const int patch_cols = 2; in TEST() local439 const int patch_cols = 2; in TEST() local505 const int patch_cols = 2; in TEST() local572 const int patch_cols = 5; in TEST() local639 const int patch_cols = 5; in TEST() local[all …]
35 const int patch_cols = 4; in TEST() local80 const int patch_cols = 4; in TEST() local200 const int patch_cols = 4; in TEST() local252 const int patch_cols = 4; in TEST() local310 const int patch_cols = 4; in TEST() local366 const int patch_cols = 3; in TEST() local414 const int patch_cols = 1; in TEST() local462 const int patch_cols = 3; in TEST() local513 const int patch_cols = 3; in TEST() local564 const int patch_cols = 3; in TEST() local[all …]
826 const int64_t patch_cols = filter.dim_size(1); in operator ()() local
174 const unsigned int patch_rows, unsigned int patch_cols, // Patch size in fill_nchw_patch_array()278 const auto patch_cols = arm_gemm::roundup<unsigned int>(output_cols, 16 / element_size); in fill_patch_array_generic_kernel() local
216 unsigned int patch_rows, unsigned int patch_cols, // Patch size in fill_nchw_patch_array()
843 extract_image_patches(const Index patch_rows, const Index patch_cols, in extract_image_patches()