Searched refs:sparse_result (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/test/ |
| D | test_sparse_semi_structured.py | 211 sparse_result = model(input)
221 assert sparse_result.stride() == sparse_compile_result.stride()
300 sparse_result = torch.mm(A_sparse, B)
304 sparse_result = torch.mm(A_sparse, B)
307 sparse_result = torch.mm(A_sparse, B)
308 torch.testing.assert_close(dense_result, sparse_result, rtol=1e-3, atol=1e-3)
330 sparse_result = torch.mm(A_sparse, B.t())
334 sparse_result = torch.mm(A_sparse, B.t())
338 sparse_result = torch.mm(A_sparse, B.t())
339 torch.testing.assert_close(dense_result, sparse_result, rtol=1e-3, atol=1e-3)
[all …]
|
| D | test_sparse.py | 1248 sparse_result = t_sparse.index_select(d, t_idx)
1249 self.assertEqual(dense_result, sparse_result)
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_string_ops.py | 519 sparse_result = string_ops.string_split_v2(
522 values=sparse_result.values,
523 value_rowids=sparse_result.indices[:, 0],
524 nrows=sparse_result.dense_shape[0],
580 sparse_result = string_ops.string_split(
583 return sparse_result
586 values=sparse_result.values,
587 value_rowids=sparse_result.indices[:, 0],
588 nrows=sparse_result.dense_shape[0],
|
| /external/tensorflow/tensorflow/python/debug/wrappers/ |
| D | local_cli_wrapper_test.py | 272 sparse_result = wrapped_sess.run(
274 self.assertAllEqual([[0, 1], [0, 2]], sparse_result.indices)
275 self.assertAllClose([20.0, 40.0], sparse_result.values)
|
| /external/pytorch/torch/csrc/distributed/c10d/ |
| D | reducer.cpp | 1632 auto sparse_result = at::sparse_coo_tensor( in finalize_backward() local
1636 bucket.gradients.copy_(sparse_result); in finalize_backward()
|