Home
last modified time | relevance | path

Searched refs:sparse_tensors (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Dcheckpoint_test_base.py65 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 Dfrom_sparse_tensor_slices_test.py187 sparse_tensors=True)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
H A Dsparse_split_op_test.py191 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 Dfeature_column_v2.py2323 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 Dfeature_column.py2067 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
2528sparse_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
2593sparse_tensors = self.categorical_column._get_sparse_tensors(inputs) # pylint: disable=protected-…
2595 sparse_tensors.id_tensor)
2650sparse_tensors = self.categorical_column._get_sparse_tensors( # pylint: disable=protected-access
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_tensors_map_ops.cc76 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 Dsparse_ops.py1043 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 Dparsing_ops.py365 sparse_tensors = [
371 sparse_tensors + dense_values + ragged_tensors))
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
H A Dcsr_sparse_matrix_ops_test.py275 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 Dtest_sparse.py4495 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 Dcommon_utils.py3335sparse_tensors = [random_sparse_compressed(n_compressed_dims, n_plain_dims, blocknnz) for _ in ran…
3336 sparse_tensors_it = map(list, zip(*sparse_tensors))