Searched defs:sparse_tensors (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | checkpoint_test_base.py | 65 def verify_unused_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
75 def verify_fully_used_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
85 def verify_exhausted_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
95 def verify_multiple_breaks(obj, ds_fn, num_outputs, sparse_tensors=False): argument
108 sparse_tensors=False): argument
138 sparse_tensors=False, argument
161 sparse_tensors=False): argument
180 def verify_exhausted_iterator(self, ds_fn, num_outputs, sparse_tensors=False): argument
210 sparse_tensors=False, argument
237 sparse_tensors=False, argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_tensors_map_ops.cc | 76 std::vector<SparseTensor>* sparse_tensors) { in RetrieveAndClearSparseTensors()
392 std::vector<SparseTensor> sparse_tensors; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/ |
| H A D | feature_column_v2.py | 2976 def _get_dense_tensor_internal_helper(self, sparse_tensors, argument
3004 def _get_dense_tensor_internal(self, sparse_tensors, state_manager): argument
3011 def _old_get_dense_tensor_internal(self, sparse_tensors, weight_collections, argument
4472 def _get_sparse_tensors_helper(self, sparse_tensors): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | sparse_ops.py | 2928 def _assert_sparse_compatible(sparse_tensors): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 4495 def __init__(self, sparse_tensors): argument
|