| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | dataset_serialization_test_base.py | 65 def run_core_tests(self, ds_fn, num_outputs, sparse_tensors=False): argument
85 ds_fn_no_opt, num_outputs, sparse_tensors=sparse_tensors)
87 ds_fn_no_opt, num_outputs, sparse_tensors=sparse_tensors)
89 ds_fn_no_opt, num_outputs, sparse_tensors=sparse_tensors)
91 ds_fn_no_opt, num_outputs, sparse_tensors=sparse_tensors)
93 ds_fn_no_opt, num_outputs, sparse_tensors=sparse_tensors)
98 sparse_tensors=False, argument
114 sparse_tensors=sparse_tensors,
118 sparse_tensors=False): argument
135 ds_fn, [num_outputs], num_outputs, sparse_tensors=sparse_tensors)
[all …]
|
| D | dataset_constructor_serialization_test.py | 95 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 | 2561 sparse_tensors = column.get_sparse_tensors(transformation_cache,
2563 id_tensor = sparse_ops.sparse_reshape(sparse_tensors.id_tensor, [
2564 array_ops.shape(sparse_tensors.id_tensor)[0], -1
2566 weight_tensor = sparse_tensors.weight_tensor
3196 def _get_dense_tensor_internal_helper(self, sparse_tensors, argument
3198 sparse_ids = sparse_tensors.id_tensor
3199 sparse_weights = sparse_tensors.weight_tensor
3224 def _get_dense_tensor_internal(self, sparse_tensors, state_manager): argument
3228 return self._get_dense_tensor_internal_helper(sparse_tensors,
3231 def _old_get_dense_tensor_internal(self, sparse_tensors, weight_collections, argument
[all …]
|
| D | feature_column.py | 2046 sparse_tensors = column._get_sparse_tensors( # pylint: disable=protected-access
2050 id_tensor = sparse_ops.sparse_reshape(sparse_tensors.id_tensor, [
2051 array_ops.shape(sparse_tensors.id_tensor)[0], -1
2053 weight_tensor = sparse_tensors.weight_tensor
2507 … sparse_tensors = self.categorical_column._get_sparse_tensors( # pylint: disable=protected-access
2509 sparse_ids = sparse_tensors.id_tensor
2510 sparse_weights = sparse_tensors.weight_tensor
2572 …sparse_tensors = self.categorical_column._get_sparse_tensors(inputs) # pylint: disable=protected-…
2574 sparse_tensors.id_tensor)
2629 … 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/python/ops/ |
| D | parsing_ops.py | 366 sparse_tensors = [
372 sparse_tensors + dense_values + ragged_tensors))
|
| D | sparse_ops.py | 909 sparse_tensors = []
911 sparse_tensors.append(
914 return sparse_tensors
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_ops_test.py | 262 sparse_tensors = list()
264 sparse_tensors.append(
270 st.indices, st.values, st.dense_shape) for st in sparse_tensors
300 st_values, st_rt_values = self.evaluate((sparse_tensors, st_rt))
|