Home
last modified time | relevance | path

Searched refs:sparse_tensor (Results 1 – 25 of 249) sorted by relevance

12345678910

/external/tensorflow/tensorflow/python/data/util/
Dsparse_test.py26 from tensorflow.python.framework import sparse_tensor
53 "classes": (ops.Tensor, sparse_tensor.SparseTensor),
57 "classes": (sparse_tensor.SparseTensor, sparse_tensor.SparseTensor),
62 "classes": (sparse_tensor.SparseTensor, ops.Tensor),
66 "classes": (((sparse_tensor.SparseTensor))),
96 "classes": sparse_tensor.SparseTensor,
106 "classes": (sparse_tensor.SparseTensor),
121 "classes": (sparse_tensor.SparseTensor, ()),
126 "classes": ((), sparse_tensor.SparseTensor),
136 "classes": (sparse_tensor.SparseTensor, (),
[all …]
Dstructure_test.py30 from tensorflow.python.framework import sparse_tensor
47 (lambda: sparse_tensor.SparseTensor(
58 "b": (sparse_tensor.SparseTensor(
60 sparse_tensor.SparseTensor(
82 (lambda: sparse_tensor.SparseTensor(
85 sparse_tensor.SparseTensor(
87 sparse_tensor.SparseTensorValue(
93 sparse_tensor.SparseTensor(
97 sparse_tensor.SparseTensor(
116 sparse_tensor.SparseTensor(
[all …]
Dsparse.py23 from tensorflow.python.framework import sparse_tensor
37 return any(c is sparse_tensor.SparseTensor for c in nest.flatten(classes))
53 tensor_shape.unknown_shape() if c is sparse_tensor.SparseTensor else shape
72 dtypes.variant if c is sparse_tensor.SparseTensor else ty
93 if c is sparse_tensor.SparseTensor else tensor
113 sparse_tensor.SparseTensor
114 if isinstance(tensor, sparse_tensor.SparseTensor) else ops.Tensor
131 if sparse_tensor.is_sparse(tensor) else tensor
149 if isinstance(tensor, sparse_tensor.SparseTensor) else tensor
/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor_test.py26 from tensorflow.python.framework import sparse_tensor
38 sp_value = sparse_tensor.SparseTensorValue(indices, values, shape)
40 sparse_tensor.SparseTensor(indices, values, shape),
41 sparse_tensor.SparseTensor.from_value(sp_value),
42 sparse_tensor.SparseTensor.from_value(
43 sparse_tensor.SparseTensor(indices, values, shape))]:
60 self.assertFalse(sparse_tensor.is_sparse(3))
61 self.assertFalse(sparse_tensor.is_sparse("foo"))
62 self.assertFalse(sparse_tensor.is_sparse(np.array(3)))
64 sparse_tensor.is_sparse(sparse_tensor.SparseTensor([[0]], [0], [1])))
[all …]
Dcomposite_tensor_test.py22 from tensorflow.python.framework import sparse_tensor
66 st1 = sparse_tensor.SparseTensor([[0, 3], [7, 2]], [1, 2], [10, 10])
67 st2 = sparse_tensor.SparseTensor([[1, 2, 3]], ['a'], [10, 10, 10])
80 st1 = sparse_tensor.SparseTensor([[0, 3], [7, 2]], [1, 2], [10, 10])
81 st2 = sparse_tensor.SparseTensor([[1, 2, 3]], ['a'], [10, 10, 10])
90 st1 = sparse_tensor.SparseTensor([[0]], [0], [100])
91 st2 = sparse_tensor.SparseTensor([[0, 3]], ['x'], [100, 100])
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_tensor_slices_test.py26 from tensorflow.python.framework import sparse_tensor
59 components = (sparse_tensor.SparseTensorValue(
63 sparse_tensor.SparseTensorValue(
75 (sparse_tensor.SparseTensorValue(
79 sparse_tensor.SparseTensorValue(
83 (sparse_tensor.SparseTensorValue(
87 sparse_tensor.SparseTensorValue(
91 (sparse_tensor.SparseTensorValue(
95 sparse_tensor.SparseTensorValue(
107 sparse_tensor.SparseTensorValue(
[all …]
Dbatch_test.py29 from tensorflow.python.framework import sparse_tensor
102 return sparse_tensor.SparseTensorValue(
107 sparse_tensor.SparseTensorValue(
117 return sparse_tensor.SparseTensorValue(
133 sparse_tensor.SparseTensorValue(
142 return sparse_tensor.SparseTensorValue(
147 sparse_tensor.SparseTensorValue(
Dfrom_sparse_tensor_slices_test.py26 from tensorflow.python.framework import sparse_tensor
41 get_next = sparse_tensor.SparseTensor(*iterator.get_next())
51 sparse_feed = sparse_tensor.SparseTensorValue(indices, values,
68 sparse_feed = sparse_tensor.SparseTensorValue(
77 sparse_feed = sparse_tensor.SparseTensorValue(empty_indices, empty_values,
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_from_sparse_op_test.py24 from tensorflow.python.framework import sparse_tensor
36 st = sparse_tensor.SparseTensor(
45 st = sparse_tensor.SparseTensor(
54 st1 = sparse_tensor.SparseTensor(indices=[[0]], values=[0], dense_shape=[3])
58 st2 = sparse_tensor.SparseTensor(
64 st3 = sparse_tensor.SparseTensor(
73 st1 = sparse_tensor.SparseTensor(
77 st2 = sparse_tensor.SparseTensor(
93 st1 = sparse_tensor.SparseTensor(
99 st2 = sparse_tensor.SparseTensor(
[all …]
/external/tensorflow/tensorflow/python/feature_column/
Dsequence_feature_column_test.py31 from tensorflow.python.framework import sparse_tensor
91 sparse_input_a = sparse_tensor.SparseTensorValue(**sparse_input_args_a)
92 sparse_input_b = sparse_tensor.SparseTensorValue(**sparse_input_args_b)
149 sparse_input = sparse_tensor.SparseTensorValue(
172 sparse_input_a = sparse_tensor.SparseTensorValue(
178 sparse_input_b = sparse_tensor.SparseTensorValue(
240 sparse_input_a = sparse_tensor.SparseTensorValue(
246 sparse_input_b = sparse_tensor.SparseTensorValue(
314 sparse_input_a = sparse_tensor.SparseTensorValue(**sparse_input_args_a)
315 sparse_input_b = sparse_tensor.SparseTensorValue(**sparse_input_args_b)
[all …]
Dfeature_column_test.py46 from tensorflow.python.framework import sparse_tensor
176 builder = _LazyBuilder(features={'a': sparse_tensor.SparseTensor(
336 sparse_tensor.SparseTensor(
539 sparse_tensor.SparseTensor(
738 sparse_tensor.SparseTensorValue(
747 wire_tensor = sparse_tensor.SparseTensor(
769 int_tensor = sparse_tensor.SparseTensor(
773 string_tensor = sparse_tensor.SparseTensor(
777 float_tensor = sparse_tensor.SparseTensor(
794 wire_tensor = sparse_tensor.SparseTensor(
[all …]
Dfeature_column_v2_test.py40 from tensorflow.python.framework import sparse_tensor
228 sparse_tensor.SparseTensor(
399 sparse_tensor.SparseTensor(
648 sparse_tensor.SparseTensor(
905 sparse_tensor.SparseTensorValue(
913 wire_tensor = sparse_tensor.SparseTensor(
939 int_tensor = sparse_tensor.SparseTensor(
941 string_tensor = sparse_tensor.SparseTensor(
943 float_tensor = sparse_tensor.SparseTensor(
958 wire_tensor = sparse_tensor.SparseTensor(
[all …]
/external/tensorflow/tensorflow/python/ops/
Dsets_impl.py24 from tensorflow.python.framework import sparse_tensor
51 a = sparse_tensor.convert_to_tensor_or_sparse_tensor(a, name="a")
52 if not isinstance(a, sparse_tensor.SparseTensor):
81 a = sparse_tensor.convert_to_tensor_or_sparse_tensor(a, name="a")
84 b = sparse_tensor.convert_to_tensor_or_sparse_tensor(b, name="b")
87 if (isinstance(a, sparse_tensor.SparseTensor) and
88 not isinstance(b, sparse_tensor.SparseTensor)):
118 if isinstance(a, sparse_tensor.SparseTensor):
119 if isinstance(b, sparse_tensor.SparseTensor):
127 elif isinstance(b, sparse_tensor.SparseTensor):
[all …]
Dsparse_ops.py33 from tensorflow.python.framework import sparse_tensor
65 if isinstance(sp_input, sparse_tensor.SparseTensorValue):
66 return sparse_tensor.SparseTensor.from_value(sp_input)
67 if not isinstance(sp_input, sparse_tensor.SparseTensor):
153 return sparse_tensor.SparseTensor(
185 return sparse_tensor.SparseTensor(
334 return sparse_tensor.SparseTensor(output_ind, output_val, output_shape)
458 sparse_classes = (sparse_tensor.SparseTensor, sparse_tensor.SparseTensorValue)
480 return sparse_tensor.SparseTensor(output_ind, output_val, output_shape)
581 isinstance(i, sparse_tensor.SparseTensor) or isinstance(i, ops.Tensor)
[all …]
/external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/
Dsequence_feature_column_test.py32 from tensorflow.python.framework import sparse_tensor
88 sparse_input_a = sparse_tensor.SparseTensorValue(**sparse_input_args_a)
89 sparse_input_b = sparse_tensor.SparseTensorValue(**sparse_input_args_b)
147 sparse_input = sparse_tensor.SparseTensorValue(
168 sparse_input_a = sparse_tensor.SparseTensorValue(
174 sparse_input_b = sparse_tensor.SparseTensorValue(
236 sparse_input_a = sparse_tensor.SparseTensorValue(
242 sparse_input_b = sparse_tensor.SparseTensorValue(
312 sparse_input_a = sparse_tensor.SparseTensorValue(**sparse_input_args_a)
313 sparse_input_b = sparse_tensor.SparseTensorValue(**sparse_input_args_b)
[all …]
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dfeature_column_ops_test.py34 from tensorflow.python.framework import sparse_tensor
58 rating_tensor = sparse_tensor.SparseTensor(
76 rating_tensor = sparse_tensor.SparseTensor(
131 wire_tensor = sparse_tensor.SparseTensor(
154 wire_tensor = sparse_tensor.SparseTensor(
182 self.assertIsInstance(output, sparse_tensor.SparseTensor)
190 wire_tensor = sparse_tensor.SparseTensor(
216 wire_tensor = sparse_tensor.SparseTensor(
247 self.assertIsInstance(output, sparse_tensor.SparseTensor)
257 wire_tensor = sparse_tensor.SparseTensor(
[all …]
/external/tensorflow/tensorflow/contrib/training/python/training/
Dbatch_sequences_with_states_test.py29 from tensorflow.python.framework import sparse_tensor
53 sp_tensor1 = sparse_tensor.SparseTensor(
68 sp_tensor2 = sparse_tensor.SparseTensor(
72 sp_tensor3 = sparse_tensor.SparseTensor(
77 self.sp_tensor3_expected = sparse_tensor.SparseTensorValue(
286 expected_seq3_batch1 = sparse_tensor.SparseTensorValue(
288 expected_seq3_batch2 = sparse_tensor.SparseTensorValue(
290 expected_seq4_batch1 = sparse_tensor.SparseTensorValue(
292 expected_seq4_batch2 = sparse_tensor.SparseTensorValue(
434 expected_seq3_batch1 = sparse_tensor.SparseTensorValue(
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_ops_test.py27 from tensorflow.python.framework import sparse_tensor
49 return sparse_tensor.SparseTensor(
59 return sparse_tensor.SparseTensor(
70 return sparse_tensor.SparseTensor(
123 indices = sparse_tensor.SparseTensorValue(
126 values = sparse_tensor.SparseTensorValue(
133 return (sparse_tensor.SparseTensor.from_value(indices),
134 sparse_tensor.SparseTensor.from_value(values))
153 sparse_tensor.SparseTensor.from_value(indices_v)):
155 sparse_tensor.SparseTensor.from_value(values_v)):
[all …]
Dsparse_serialization_ops_test.py24 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib
259 sparse_tensor = self._SparseTensorPlaceholder(dtype=dtypes.string)
260 serialized = serialize_many_fn(sparse_tensor, out_type=out_type)
264 sparse_tensor.indices: indices_value,
265 sparse_tensor.values: values_value,
266 sparse_tensor.dense_shape: shape_value
288 sparse_tensor = self._SparseTensorPlaceholder(dtype=dtypes.string)
289 serialized = serialize_many_fn(sparse_tensor, out_type=out_type)
294 sparse_tensor.indices: indices_value,
295 sparse_tensor.values: values_value,
[all …]
Dsparse_concat_op_test.py25 from tensorflow.python.framework import sparse_tensor
38 return sparse_tensor.SparseTensor(
53 return sparse_tensor.SparseTensorValue(
58 return sparse_tensor.SparseTensor.from_value(self._SparseTensorValue_3x3())
67 return sparse_tensor.SparseTensorValue(
72 return sparse_tensor.SparseTensor.from_value(self._SparseTensorValue_3x5())
81 return sparse_tensor.SparseTensor(
92 return sparse_tensor.SparseTensor(
106 return sparse_tensor.SparseTensor(
118 return sparse_tensor.SparseTensor(
[all …]
/external/tensorflow/tensorflow/contrib/layers/python/ops/
Dsparse_ops_test.py25 from tensorflow.python.framework import sparse_tensor
159 sparse_input = sparse_tensor.SparseTensor(
171 sparse_input = sparse_tensor.SparseTensor(
183 sparse_input = sparse_tensor.SparseTensor(
195 sparse_input = sparse_tensor.SparseTensor(
211 _assert_sparse_tensor_value(self, sparse_tensor.SparseTensorValue(
224 _assert_sparse_tensor_value(self, sparse_tensor.SparseTensorValue(
239 _assert_sparse_tensor_value(self, sparse_tensor.SparseTensorValue(
259 _assert_sparse_tensor_value(self, sparse_tensor.SparseTensorValue(
273 _assert_sparse_tensor_value(self, sparse_tensor.SparseTensorValue(
[all …]
Dsparse_feature_cross_op.py25 from tensorflow.python.framework import sparse_tensor
72 if not all(isinstance(i, sparse_tensor.SparseTensor) or
77 if isinstance(i, sparse_tensor.SparseTensor)]
79 if not isinstance(i, sparse_tensor.SparseTensor)]
122 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
/external/tensorflow/tensorflow/contrib/data/python/kernel_tests/
Dslide_dataset_op_test.py28 from tensorflow.python.framework import sparse_tensor
206 return sparse_tensor.SparseTensorValue(
220 expected = sparse_tensor.SparseTensorValue(
224 self.assertTrue(sparse_tensor.is_sparse(actual))
232 return sparse_tensor.SparseTensorValue(
255 expected = sparse_tensor.SparseTensorValue(
259 self.assertTrue(sparse_tensor.is_sparse(actual))
267 return sparse_tensor.SparseTensorValue(
281 expected = sparse_tensor.SparseTensorValue(
287 self.assertTrue(sparse_tensor.is_sparse(actual))
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dbatch_features.cc100 sparse::SparseTensor sparse_tensor; in Initialize() local
103 order_dims, &sparse_tensor)); in Initialize()
104 sparse_float_feature_columns_.push_back(std::move(sparse_tensor)); in Initialize()
142 sparse::SparseTensor sparse_tensor; in Initialize() local
145 order_dims, &sparse_tensor)); in Initialize()
146 sparse_int_feature_columns_.push_back(std::move(sparse_tensor)); in Initialize()
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/
Dsdca_estimator_test.py27 from tensorflow.python.framework import sparse_tensor
123 sparse_tensor.SparseTensor(
151 sparse_tensor.SparseTensor(
156 sparse_tensor.SparseTensor(
182 sparse_tensor.SparseTensor(
188 sparse_tensor.SparseTensor(
214 sparse_tensor.SparseTensor(
219 sparse_tensor.SparseTensor(
250 sparse_tensor.SparseTensor(
289 sparse_tensor.SparseTensor(
[all …]

12345678910