| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | dataset_serialization_test_base.py | 64 def run_core_tests(self, ds_fn1, ds_fn2, num_outputs, sparse_tensors=False): argument
86 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
88 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
90 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
92 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
94 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
96 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
98 ds_fn1_no_opt, num_outputs, sparse_tensors=sparse_tensors)
108 sparse_tensors=sparse_tensors)
113 sparse_tensors=False, argument
[all …]
|
| D | dataset_constructor_serialization_test.py | 91 sparse_tensors=True)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_split_op_test.py | 200 sparse_tensors = sparse_ops.sparse_split(
202 self.assertAllEqual(len(sparse_tensors), 3)
203 self.assertAllEqual(sparse_tensors[0].indices.eval(), [[0, 0], [1, 1],
205 self.assertAllEqual(sparse_tensors[0].values.eval(), [0, 11, 20, 30])
206 self.assertAllEqual(sparse_tensors[0].dense_shape.eval(), [4, 2])
207 self.assertAllEqual(sparse_tensors[1].indices.eval(),
209 self.assertAllEqual(sparse_tensors[1].values.eval(), [2, 13, 23, 32, 33])
210 self.assertAllEqual(sparse_tensors[1].dense_shape.eval(), [4, 2])
211 self.assertAllEqual(sparse_tensors[2].indices.eval(),
213 self.assertAllEqual(sparse_tensors[2].values.eval(), [4, 5, 14, 25, 35])
[all …]
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 2445 sparse_tensors = column.get_sparse_tensors(transformation_cache,
2447 id_tensor = sparse_ops.sparse_reshape(sparse_tensors.id_tensor, [
2448 array_ops.shape(sparse_tensors.id_tensor)[0], -1
2450 weight_tensor = sparse_tensors.weight_tensor
3049 def _get_dense_tensor_internal_helper(self, sparse_tensors, argument
3051 sparse_ids = sparse_tensors.id_tensor
3052 sparse_weights = sparse_tensors.weight_tensor
3071 def _get_dense_tensor_internal(self, sparse_tensors, state_manager): argument
3075 return self._get_dense_tensor_internal_helper(sparse_tensors,
3078 def _old_get_dense_tensor_internal(self, sparse_tensors, weight_collections, argument
[all …]
|
| D | feature_column.py | 1988 sparse_tensors = column._get_sparse_tensors( # pylint: disable=protected-access
1992 id_tensor = sparse_ops.sparse_reshape(sparse_tensors.id_tensor, [
1993 array_ops.shape(sparse_tensors.id_tensor)[0], -1
1995 weight_tensor = sparse_tensors.weight_tensor
2426 … sparse_tensors = self.categorical_column._get_sparse_tensors( # pylint: disable=protected-access
2428 sparse_ids = sparse_tensors.id_tensor
2429 sparse_weights = sparse_tensors.weight_tensor
2485 …sparse_tensors = self.categorical_column._get_sparse_tensors(inputs) # pylint: disable=protected-…
2487 sparse_tensors.id_tensor)
2541 … sparse_tensors = self.categorical_column._get_sparse_tensors( # pylint: disable=protected-access
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_tensors_map_ops.cc | 81 std::vector<SparseTensor>* sparse_tensors) { in RetrieveAndClearSparseTensors() argument
82 sparse_tensors->clear(); in RetrieveAndClearSparseTensors()
83 sparse_tensors->reserve(handles.size()); in RetrieveAndClearSparseTensors()
98 sparse_tensors->push_back(std::move(tensor)); in RetrieveAndClearSparseTensors()
380 std::vector<SparseTensor> sparse_tensors; in Compute() local
383 context, sparse_handles_t, &sparse_tensors)); in Compute()
386 const SparseTensor& st = sparse_tensors[i]; in Compute()
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | feature_column_ops.py | 328 sparse_tensors = []
333 sparse_tensors.append(
337 sparse_tensor = sparse_ops.sparse_concat(1, sparse_tensors)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_ops.py | 871 sparse_tensors = [
875 return dict(zip(sparse_keys + dense_keys, sparse_tensors + dense_values))
2124 sparse_tensors = [
2130 return dict(zip(sparse_keys + dense_keys, sparse_tensors + dense_values))
|
| D | sparse_ops.py | 858 sparse_tensors = []
860 sparse_tensors.append(
863 return sparse_tensors
|