Searched defs:dim_k (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseTensorMath.cpp | 1186 void s_addmm_out_sparse_dense_worker(int64_t nnz, int64_t dim_i, int64_t dim_j, int64_t dim_k, Tens… in s_addmm_out_sparse_dense_worker()
1282 int64_t dim_k = dense.size(1); in s_addmm_out_sparse_dense_cpu() local
1511 int64_t dim_k = dense.size(1); in _sspaddmm_out_cpu() local
1978 int64_t dim_k = mat2.size(2); in bmm_out_sparse_cpu() local
|
| D | SparseCsrTensorMath.cpp | 532 auto dim_k = dense.size(1); in addmm_out_sparse_csr_native_cpu() local
|
| /external/pytorch/test/ |
| D | test_sparse.py | 1368 def test_shape(num_mats, dim_i, dim_j, dim_k, nnz): argument
1424 def test_shape(num_mats, dim_i, dim_j, dim_k, nnz): argument
|