| /external/tensorflow/tensorflow/python/framework/ |
| sparse_tensor_test.py | 16 """Tests for tensorflow.python.framework.sparse_tensor."""
26 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))
[all...] |
| composite_tensor_test.py | 22 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/util/ |
| sparse_test.py | 26 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)
[all...] |
| sparse.py | 23 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
|
| structure_test.py | 30 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...] |
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| from_tensor_slices_test.py | 26 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...] |
| from_sparse_tensor_slices_test.py | 26 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,
|
| batch_test.py | 29 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(
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| ragged_from_sparse_op_test.py | 24 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/ops/ |
| sets_impl.py | 24 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...] |
| sparse_ops.py | 33 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
[all...] |
| /external/tensorflow/tensorflow/contrib/libsvm/python/ops/ |
| libsvm_ops.py | 23 from tensorflow.python.framework import sparse_tensor
47 return sparse_tensor.SparseTensor(indices, values, shape), labels
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| sequence_feature_column_test.py | 31 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...] |
| feature_column_test.py | 46 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(
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| batch_features.cc | 100 sparse::SparseTensor sparse_tensor; local
103 order_dims, &sparse_tensor));
104 sparse_float_feature_columns_.push_back(std::move(sparse_tensor));
142 sparse::SparseTensor sparse_tensor; local
145 order_dims, &sparse_tensor));
146 sparse_int_feature_columns_.push_back(std::move(sparse_tensor));
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| prettyprint_ops_test.py | 25 from tensorflow.python.framework import sparse_tensor
41 a = sparse_tensor.SparseTensor(
43 b = sparse_tensor.SparseTensor(
|
| /external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/ |
| sequence_feature_column_test.py | 32 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/python/kernel_tests/ |
| sparse_serialization_ops_test.py | 24 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...] |
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| batch_benchmark.py | 24 from tensorflow.python.framework import sparse_tensor
36 tensor = sparse_tensor.SparseTensor(
|
| /external/tensorflow/tensorflow/contrib/layers/python/ops/ |
| sparse_ops_test.py | 25 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...] |
| sparse_feature_cross_op.py | 25 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/ |
| slide_dataset_op_test.py | 28 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/layers/python/layers/ |
| encoders.py | 24 from tensorflow.python.framework import sparse_tensor
77 if isinstance(ids, sparse_tensor.SparseTensor):
84 if isinstance(ids, sparse_tensor.SparseTensor):
|
| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_slice_op.cc | 21 #include "tensorflow/core/util/sparse/sparse_tensor.h"
69 sparse::SparseTensor sparse_tensor; variable
73 TensorShape(input_shape.vec<int64>()), &sparse_tensor));
81 sparse::SparseTensor::Slice<T>(sparse_tensor, start, size);
|
| sparse_split_op.cc | 21 #include "tensorflow/core/util/sparse/sparse_tensor.h"
66 sparse::SparseTensor sparse_tensor; variable
70 TensorShape(input_shape.vec<int64>()), &sparse_tensor));
74 sparse::SparseTensor::Split<T>(sparse_tensor, split_dim,
|
