Searched defs:n_batch_dim (Results 1 – 2 of 2) sorted by relevance
81 const int64_t& n_batch_dim) { in dense_to_sparse_compressed_prepare_check_mask_values_batched()181 const int64_t& n_batch_dim, in reshape_2d_sparse_compressed_members_to_nd_batched()882 auto n_batch_dim = self.dim() - 2 - dense_dim; in _batch_tile_tensor() local1051 auto n_batch_dim = self.dim() - 2 - dense_dim; in dense_to_sparse_compressed() local
3007 const auto n_batch_dim = compressed_inds.dim() - 1; in sparse_compressed_transpose() local