Searched defs:batch_ndim (Results 1 – 4 of 4) sorted by relevance
643 auto batch_ndim = sparse_csr::numBatchDimensions(self); in sparse_compressed_to_dense() local1558 int64_t batch_ndim = crow_indices.dim() - 1; in convert_indices_from_csr_to_coo_cpu() local1875 const auto batch_ndim = compressed_indices.dim() - 1; in sparse_compressed_to_sparse() local
157 const auto batch_ndim = compressed_indices.dim() - 1; in _validate_sparse_compressed_tensor_args_worker() local527 const auto batch_ndim = compressed_indices.dim() - 1; in _estimate_sparse_compressed_tensor_size() local
108 int64_t batch_ndim = crow_indices.dim() - 1; in convert_indices_from_csr_to_coo_cuda() local
1350 def get_dense_shape(shape, batch_ndim): argument1357 def transpose(t, batch_ndim): argument