Searched defs:dense_tensor (Results 1 – 4 of 4) sorted by relevance
217 auto *dense_tensor = ctx.Input(3); in ComputeRowAndCol2() local283 auto *dense_tensor = ctx.Input(3); in SparseAddmmCompute() local
437 def construct(self, indptr, indices, values, dense_tensor, dense_vector): argument792 def test_to_csr(dense_tensor): argument
403 def test_to_coo(dense_tensor): argument
298 std::vector<int64_t> dense_tensor = dense_list_in[i]; in GenerateColumnsFromInput() local