Searched defs:first_dims (Results 1 – 2 of 2) sorted by relevance
17 int64_t first_dims = first.dim(); in check_cat_shape_except_dim() local
477 auto first_dims = IntArrayRef(sorted_dims).slice(sorted_dims.size() - max_dims, max_dims); in _fft_c2c_cufft() local