/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 69 sparse_tensors=sparse_tensors) 75 def verify_fully_used_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument 79 sparse_tensors=sparse_tensors) 85 def verify_exhausted_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument 89 sparse_tensors=sparse_tensors) 95 def verify_multiple_breaks(obj, ds_fn, num_outputs, sparse_tensors=False): argument 99 sparse_tensors=sparse_tensors) 108 sparse_tensors=False): argument 112 sparse_tensors=sparse_tensors) [all …]
|
H A D | from_sparse_tensor_slices_test.py | 187 sparse_tensors=True)
|
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
H A D | sparse_split_op_test.py | 191 sparse_tensors = self.evaluate( 194 self.assertAllEqual(len(sparse_tensors), 3) 195 self.assertAllEqual(sparse_tensors[0].indices, 197 self.assertAllEqual(sparse_tensors[0].values, [0, 11, 20, 30]) 198 self.assertAllEqual(sparse_tensors[0].dense_shape, [4, 2]) 199 self.assertAllEqual(sparse_tensors[1].indices, 201 self.assertAllEqual(sparse_tensors[1].values, [2, 13, 23, 32, 33]) 202 self.assertAllEqual(sparse_tensors[1].dense_shape, [4, 2]) 203 self.assertAllEqual(sparse_tensors[2].indices, 205 self.assertAllEqual(sparse_tensors[2].values, [4, 5, 14, 25, 35]) [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/ |
H A D | feature_column_v2.py | 2323 sparse_tensors = column.get_sparse_tensors(transformation_cache, 2325 id_tensor = sparse_ops.sparse_reshape(sparse_tensors.id_tensor, [ 2326 array_ops.shape(sparse_tensors.id_tensor)[0], -1 2328 weight_tensor = sparse_tensors.weight_tensor 2976 def _get_dense_tensor_internal_helper(self, sparse_tensors, argument 2978 sparse_ids = sparse_tensors.id_tensor 2979 sparse_weights = sparse_tensors.weight_tensor 3004 def _get_dense_tensor_internal(self, sparse_tensors, state_manager): argument 3008 return self._get_dense_tensor_internal_helper(sparse_tensors, 3011 def _old_get_dense_tensor_internal(self, sparse_tensors, weight_collections, argument [all …]
|
H A D | feature_column.py | 2067 sparse_tensors = column._get_sparse_tensors( # pylint: disable=protected-access 2071 id_tensor = sparse_ops.sparse_reshape(sparse_tensors.id_tensor, [ 2072 array_ops.shape(sparse_tensors.id_tensor)[0], -1 2074 weight_tensor = sparse_tensors.weight_tensor 2528 … sparse_tensors = self.categorical_column._get_sparse_tensors( # pylint: disable=protected-access 2530 sparse_ids = sparse_tensors.id_tensor 2531 sparse_weights = sparse_tensors.weight_tensor 2593 …sparse_tensors = self.categorical_column._get_sparse_tensors(inputs) # pylint: disable=protected-… 2595 sparse_tensors.id_tensor) 2650 … sparse_tensors = self.categorical_column._get_sparse_tensors( # pylint: disable=protected-access [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() argument 77 sparse_tensors->clear(); in RetrieveAndClearSparseTensors() 78 sparse_tensors->reserve(handles.size()); in RetrieveAndClearSparseTensors() 93 sparse_tensors->push_back(std::move(tensor)); in RetrieveAndClearSparseTensors() 392 std::vector<SparseTensor> sparse_tensors; in Compute() local 395 context, sparse_handles_t, &sparse_tensors)); in Compute() 398 const SparseTensor& st = sparse_tensors[i]; in Compute()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | sparse_ops.py | 1043 sparse_tensors = [] 1045 sparse_tensors.append( 1048 return sparse_tensors 2928 def _assert_sparse_compatible(sparse_tensors): argument 2938 first = sparse_tensors[0] 2939 for t in sparse_tensors[1:]:
|
H A D | parsing_ops.py | 365 sparse_tensors = [ 371 sparse_tensors + dense_values + ragged_tensors))
|
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
H A D | csr_sparse_matrix_ops_test.py | 275 sparse_tensors = list() 277 sparse_tensors.append( 283 st.indices, st.values, st.dense_shape) for st in sparse_tensors 313 st_values, st_rt_values = self.evaluate((sparse_tensors, st_rt))
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_sparse.py | 4495 def __init__(self, sparse_tensors): argument 4496 self.sparse_tensors = sparse_tensors 4499 return len(self.sparse_tensors) 4502 return self.sparse_tensors[index]
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_utils.py | 3335 …sparse_tensors = [random_sparse_compressed(n_compressed_dims, n_plain_dims, blocknnz) for _ in ran… 3336 sparse_tensors_it = map(list, zip(*sparse_tensors))
|