| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | dataset_serialization_test_base.py | 64 def run_core_tests(self, ds_fn, num_outputs, sparse_tensors=False): argument
84 ds_fn_no_opt, num_outputs, sparse_tensors=sparse_tensors)
86 ds_fn_no_opt, num_outputs, sparse_tensors=sparse_tensors)
88 ds_fn_no_opt, num_outputs, sparse_tensors=sparse_tensors)
90 ds_fn_no_opt, num_outputs, sparse_tensors=sparse_tensors)
92 ds_fn_no_opt, num_outputs, sparse_tensors=sparse_tensors)
97 sparse_tensors=False, argument
113 sparse_tensors=sparse_tensors,
117 sparse_tensors=False): argument
134 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 | 189 sparse_tensors = self.evaluate(
192 self.assertAllEqual(len(sparse_tensors), 3)
193 self.assertAllEqual(sparse_tensors[0].indices,
195 self.assertAllEqual(sparse_tensors[0].values, [0, 11, 20, 30])
196 self.assertAllEqual(sparse_tensors[0].dense_shape, [4, 2])
197 self.assertAllEqual(sparse_tensors[1].indices,
199 self.assertAllEqual(sparse_tensors[1].values, [2, 13, 23, 32, 33])
200 self.assertAllEqual(sparse_tensors[1].dense_shape, [4, 2])
201 self.assertAllEqual(sparse_tensors[2].indices,
203 self.assertAllEqual(sparse_tensors[2].values, [4, 5, 14, 25, 35])
[all …]
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 2259 sparse_tensors = column.get_sparse_tensors(transformation_cache,
2261 id_tensor = sparse_ops.sparse_reshape(sparse_tensors.id_tensor, [
2262 array_ops.shape(sparse_tensors.id_tensor)[0], -1
2264 weight_tensor = sparse_tensors.weight_tensor
2909 def _get_dense_tensor_internal_helper(self, sparse_tensors, argument
2911 sparse_ids = sparse_tensors.id_tensor
2912 sparse_weights = sparse_tensors.weight_tensor
2937 def _get_dense_tensor_internal(self, sparse_tensors, state_manager): argument
2941 return self._get_dense_tensor_internal_helper(sparse_tensors,
2944 def _old_get_dense_tensor_internal(self, sparse_tensors, weight_collections, argument
[all …]
|
| D | feature_column.py | 2071 sparse_tensors = column._get_sparse_tensors( # pylint: disable=protected-access
2075 id_tensor = sparse_ops.sparse_reshape(sparse_tensors.id_tensor, [
2076 array_ops.shape(sparse_tensors.id_tensor)[0], -1
2078 weight_tensor = sparse_tensors.weight_tensor
2532 … sparse_tensors = self.categorical_column._get_sparse_tensors( # pylint: disable=protected-access
2534 sparse_ids = sparse_tensors.id_tensor
2535 sparse_weights = sparse_tensors.weight_tensor
2597 …sparse_tensors = self.categorical_column._get_sparse_tensors(inputs) # pylint: disable=protected-…
2599 sparse_tensors.id_tensor)
2654 … 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()
381 std::vector<SparseTensor> sparse_tensors; in Compute() local
384 context, sparse_handles_t, &sparse_tensors)); in Compute()
387 const SparseTensor& st = sparse_tensors[i]; in Compute()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 1050 sparse_tensors = []
1052 sparse_tensors.append(
1055 return sparse_tensors
2932 def _assert_sparse_compatible(sparse_tensors): argument
2942 first = sparse_tensors[0]
2943 for t in sparse_tensors[1:]:
|
| D | parsing_ops.py | 369 sparse_tensors = [
375 sparse_tensors + dense_values + ragged_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))
|