Searched defs:num_matrices (Results 1 – 4 of 4) sorted by relevance
94 const int64 num_matrices = output.size(); in operator ()() local117 const int64 num_matrices = sign.size(); in operator ()() local
678 int64_t num_matrices, in search_end_matrix_indices_cuda_kernel()717 void search_end_matrix_indices(int64_t* mat_el_end_indices, int64_t num_matrices, const Tensor& ind… in search_end_matrix_indices()
1986 int64_t num_matrices = self_coalesced.size(0); in bmm_out_sparse_cpu() local
2244 void _fill_matrix_powers(Tensor& buffer, const Tensor& a, int num_matrices) { in _fill_matrix_powers()