Searched defs:valid_dims (Results 1 – 4 of 4) sorted by relevance
180 bool valid_dims = input.size(1) != 0 && input.size(2) != 0; in im2col_shape_check() local
143 bool valid_dims = input.size(1) != 0 && input.size(2) != 0; in pool2d_shape_check() local
217 const std::vector<int64_t> valid_dims = {0, 1, -1, -2}; in TEST_F() local
314 bool valid_dims = input.size(1) != 0 && input.size(2) != 0 && input.size(3) != 0; in shapeAndGradOutputCheck3d() local