Searched defs:nbatches (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | SparseCsrTensorMath.cu | 87 …data_out, const input_t* data_in, const int64_t nrows, const int64_t nnz, const int64_t nbatches) { in convert_indices_from_csr_to_coo_cuda_kernel()
125 int64_t nbatches = total_nnz / nnz; in convert_indices_from_csr_to_coo_cuda() local
|
| /external/pytorch/torch/sparse/ |
| D | _triton_ops.py | 892 indices_format, M, K, N, Ms, Ks, nbatches, SPLIT_N, compressed_sparse_tensor_as_key argument
2044 nbatches, argument
|