Searched defs:sparse_tensor (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/ |
| H A D | utils.py | 8 def sparse_tensor_to_rpc_format(sparse_tensor): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_slice_op.cc | 42 sparse::SparseTensor sparse_tensor; in operator ()() local
|
| H A D | sparse_split_op.cc | 39 sparse::SparseTensor sparse_tensor; in operator ()() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | sparse_tensor_slice_dataset_op.cc | 34 const sparse::SparseTensor& sparse_tensor) in Dataset()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 2247 def _test_empty_like(self, sparse_tensor, dtype, device, coalesced): argument
2350 def _test_log1p_tensor(self, sparse_tensor, coalesced): argument
2419 def _test_neg_negative(self, sparse_tensor): argument
2482 def _test_asin_arcsin(self, sparse_tensor, coalesced): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | dataset_ops.py | 3882 def from_sparse_tensor_slices(sparse_tensor): argument
4747 def __init__(self, sparse_tensor): argument
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_misc.cpp | 2402 auto sparse_tensor = i == 0 ? at::ones(10).to_sparse() in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | model_builder.cc | 815 TensorFloat32 sparse_tensor; in Parse() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | rpc_test.py | 1090 def _test_cuda_future_extraction(self, wrapper, unwrapper, sparse_tensor): argument
|