Home
last modified time | relevance | path

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

1234567891011

/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor_test.py29 from tensorflow.python.framework import sparse_tensor
45 sp_value = sparse_tensor.SparseTensorValue(indices, values, shape)
47 sparse_tensor.SparseTensor(indices, values, shape),
48 sparse_tensor.SparseTensor.from_value(sp_value),
49 sparse_tensor.SparseTensor.from_value(
50 sparse_tensor.SparseTensor(indices, values, shape))]:
73 tensor = sparse_tensor.SparseTensor(
78 self.assertFalse(sparse_tensor.is_sparse(3))
79 self.assertFalse(sparse_tensor.is_sparse("foo"))
80 self.assertFalse(sparse_tensor.is_sparse(np.array(3)))
[all …]
/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),
140 (sparse_tensor.SparseTensor, (), sparse_tensor.SparseTensor),
[all …]
Dstructure_test.py34 from tensorflow.python.framework import sparse_tensor
63 ("SparseTensor", lambda: sparse_tensor.SparseTensor(
65 sparse_tensor.SparseTensorSpec, [dtypes.variant], [None]),
78 "b": (sparse_tensor.SparseTensor(
80 sparse_tensor.SparseTensor(
120 ("SparseTensor", lambda: sparse_tensor.SparseTensor(
123 sparse_tensor.SparseTensor(
125 sparse_tensor.SparseTensorValue(
131 sparse_tensor.SparseTensor(
135 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/ops/ragged/
Dragged_from_sparse_op_test.py24 from tensorflow.python.framework import sparse_tensor
35 st = sparse_tensor.SparseTensor(
44 st = sparse_tensor.SparseTensor(
53 st1 = sparse_tensor.SparseTensor(indices=[[0]], values=[0], dense_shape=[3])
57 st2 = sparse_tensor.SparseTensor(
63 st3 = sparse_tensor.SparseTensor(
72 st1 = sparse_tensor.SparseTensor(
76 st2 = sparse_tensor.SparseTensor(
92 st1 = sparse_tensor.SparseTensor(
98 st2 = sparse_tensor.SparseTensor(
[all …]
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_tensor_slices_test.py30 from tensorflow.python.framework import sparse_tensor
90 components = (sparse_tensor.SparseTensorValue(
94 sparse_tensor.SparseTensorValue(
106 (sparse_tensor.SparseTensorValue(
110 sparse_tensor.SparseTensorValue(
114 (sparse_tensor.SparseTensorValue(
118 sparse_tensor.SparseTensorValue(
122 (sparse_tensor.SparseTensorValue(
126 sparse_tensor.SparseTensorValue(
139 sparse_tensor.SparseTensorValue(
[all …]
Dfrom_sparse_tensor_slices_test.py28 from tensorflow.python.framework import sparse_tensor
45 get_next = sparse_tensor.SparseTensor(*iterator.get_next())
55 sparse_feed = sparse_tensor.SparseTensorValue(indices, values,
72 sparse_feed = sparse_tensor.SparseTensorValue(
81 sparse_feed = sparse_tensor.SparseTensorValue(empty_indices, empty_values,
/external/tensorflow/tensorflow/python/ops/
Dsets_impl.py24 from tensorflow.python.framework import sparse_tensor
53 a = sparse_tensor.convert_to_tensor_or_sparse_tensor(a, name="a")
54 if not isinstance(a, sparse_tensor.SparseTensor):
83 a = sparse_tensor.convert_to_tensor_or_sparse_tensor(a, name="a")
86 b = sparse_tensor.convert_to_tensor_or_sparse_tensor(b, name="b")
89 if (isinstance(a, sparse_tensor.SparseTensor) and
90 not isinstance(b, sparse_tensor.SparseTensor)):
120 if isinstance(a, sparse_tensor.SparseTensor):
121 if isinstance(b, sparse_tensor.SparseTensor):
129 elif isinstance(b, sparse_tensor.SparseTensor):
[all …]
Dsparse_ops.py34 from tensorflow.python.framework import sparse_tensor
69 if isinstance(sp_input, sparse_tensor.SparseTensorValue):
70 return sparse_tensor.SparseTensor.from_value(sp_input)
71 if not isinstance(sp_input, sparse_tensor.SparseTensor):
136 return sparse_tensor.SparseTensor(indices, values, shape)
241 return sparse_tensor.SparseTensor(
273 return sparse_tensor.SparseTensor(
440 return sparse_tensor.SparseTensor(output_ind, output_val, output_shape)
445 output = sparse_tensor.SparseTensor(output_ind, output_val, output_shape)
573 sparse_classes = (sparse_tensor.SparseTensor, sparse_tensor.SparseTensorValue)
[all …]
Dbincount_ops.py24 from tensorflow.python.framework import sparse_tensor
143 if not isinstance(arr, sparse_tensor.SparseTensor):
146 if not isinstance(weights, sparse_tensor.SparseTensor):
166 if isinstance(arr, sparse_tensor.SparseTensor):
182 if isinstance(arr, sparse_tensor.SparseTensor):
195 if isinstance(arr, sparse_tensor.SparseTensor):
388 if not isinstance(values, sparse_tensor.SparseTensor):
392 if not isinstance(weights, sparse_tensor.SparseTensor):
410 if isinstance(values, sparse_tensor.SparseTensor):
423 if isinstance(values, sparse_tensor.SparseTensor):
[all …]
Dfunctional_ops_test.py25 from tensorflow.python.framework import sparse_tensor
71 return sparse_tensor.SparseTensor([[0]], [x + 1], [1])
75 return sparse_tensor.SparseTensor([[0]], [x - 1], [1])
81 self.assertIsInstance(result, sparse_tensor.SparseTensor)
/external/tensorflow/tensorflow/python/feature_column/
Dsequence_feature_column_test.py33 from tensorflow.python.framework import sparse_tensor
173 inputs = sparse_tensor.SparseTensorValue(**inputs_args)
174 expected = sparse_tensor.SparseTensorValue(**expected_args)
247 inputs = sparse_tensor.SparseTensorValue(**inputs_args)
248 expected = sparse_tensor.SparseTensorValue(**expected_args)
298 inputs = sparse_tensor.SparseTensorValue(**inputs_args)
299 expected = sparse_tensor.SparseTensorValue(**expected_args)
316 inputs = sparse_tensor.SparseTensorValue(
318 expected = sparse_tensor.SparseTensorValue(
366 inputs = sparse_tensor.SparseTensorValue(**inputs_args)
[all …]
Dfeature_column_test.py47 from tensorflow.python.framework import sparse_tensor
176 builder = _LazyBuilder(features={'a': sparse_tensor.SparseTensor(
333 sparse_tensor.SparseTensor(
529 sparse_tensor.SparseTensor(
725 sparse_tensor.SparseTensorValue(
732 wire_tensor = sparse_tensor.SparseTensor(
753 int_tensor = sparse_tensor.SparseTensor(
757 string_tensor = sparse_tensor.SparseTensor(
761 float_tensor = sparse_tensor.SparseTensor(
778 wire_tensor = sparse_tensor.SparseTensor(
[all …]
Dfeature_column_v2_test.py41 from tensorflow.python.framework import sparse_tensor
248 sparse_tensor.SparseTensor(
416 sparse_tensor.SparseTensor(
648 sparse_tensor.SparseTensor(
849 sparse_tensor.SparseTensorValue(
856 wire_tensor = sparse_tensor.SparseTensor(
882 int_tensor = sparse_tensor.SparseTensor(
884 string_tensor = sparse_tensor.SparseTensor(
886 float_tensor = sparse_tensor.SparseTensor(
901 wire_tensor = sparse_tensor.SparseTensor(
[all …]
/external/tensorflow/tensorflow/python/keras/feature_column/
Dsequence_feature_column_test.py33 from tensorflow.python.framework import sparse_tensor
98 sparse_input_a = sparse_tensor.SparseTensorValue(**sparse_input_args_a)
99 sparse_input_b = sparse_tensor.SparseTensorValue(**sparse_input_args_b)
156 sparse_input = sparse_tensor.SparseTensorValue(
177 sparse_input_a = sparse_tensor.SparseTensorValue(
183 sparse_input_b = sparse_tensor.SparseTensorValue(
245 sparse_input_a = sparse_tensor.SparseTensorValue(
251 sparse_input_b = sparse_tensor.SparseTensorValue(
319 sparse_input_a = sparse_tensor.SparseTensorValue(**sparse_input_args_a)
320 sparse_input_b = sparse_tensor.SparseTensorValue(**sparse_input_args_b)
[all …]
Ddense_features_test.py32 from tensorflow.python.framework import sparse_tensor
63 sparse_input = sparse_tensor.SparseTensor(
108 sparse_input = sparse_tensor.SparseTensor(
160 sparse_input = sparse_tensor.SparseTensor(
368 sparse_tensor.SparseTensor(
471 sparse_tensor.SparseTensor(
476 sparse_tensor.SparseTensor(
506 sparse_tensor.SparseTensor(
511 sparse_tensor.SparseTensor(
524 sparse_tensor.SparseTensor(
[all …]
/external/tensorflow/tensorflow/core/util/sparse/
DBUILD20 "sparse_tensor.cc",
21 "sparse_tensor.h",
30 "sparse_tensor.h",
40 "sparse_tensor.h",
49 "sparse_tensor.cc",
/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 …]
Dsparse_cross_op_test.py27 from tensorflow.python.framework import sparse_tensor
62 return sparse_tensor.SparseTensor(
455 return sparse_tensor.SparseTensor(
465 st1 = sparse_tensor.SparseTensor([[0, 0]], [0], [2, 2])
471 st2 = sparse_tensor.SparseTensor([[0, 0]], [0], [2, 2])
477 st3 = sparse_tensor.SparseTensor([[0, 0]], [0], [2, 2])
487 st = sparse_tensor.SparseTensor([[0]], [0], [10]) # 1D SparseTensor
496 st1 = sparse_tensor.SparseTensor([[0, 0]], [0], [10, 10]) # batch size 10
497 st2 = sparse_tensor.SparseTensor([[0, 0]], [0], [7, 10]) # batch size 7
522 out = sparse_tensor.SparseTensor(inds, vals, shapes)
[all …]
Dsparse_reorder_op_test.py24 from tensorflow.python.framework import sparse_tensor
36 return sparse_tensor.SparseTensor(
50 return sparse_tensor.SparseTensorValue(ind, val, shape)
53 sp_input = sparse_tensor.SparseTensor.from_value(
114 sp_input = sparse_tensor.SparseTensor(input_val.indices,
129 sp_input = sparse_tensor.SparseTensor(
Dsparse_reshape_op_test.py27 from tensorflow.python.framework import sparse_tensor
37 return sparse_tensor.SparseTensor(
47 return sparse_tensor.SparseTensorValue(ind, val, shape)
54 return sparse_tensor.SparseTensorValue(ind, val, shape)
57 sp_input = sparse_tensor.SparseTensor.from_value(
64 sp_input = sparse_tensor.SparseTensor.from_value(
72 sp_input = sparse_tensor.SparseTensor.from_value(
79 sp_input = sparse_tensor.SparseTensor.from_value(
86 sp_input = sparse_tensor.SparseTensor.from_value(
251 sp_input = sparse_tensor.SparseTensor.from_value(input_val)
[all …]
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dcategory_crossing_test.py25 from tensorflow.python.framework import sparse_tensor
45 inputs_0 = sparse_tensor.SparseTensor(
49 inputs_1 = sparse_tensor.SparseTensor(
57 inputs_0 = sparse_tensor.SparseTensor(
61 inputs_1 = sparse_tensor.SparseTensor(
69 inputs_0 = sparse_tensor.SparseTensor(
73 inputs_1 = sparse_tensor.SparseTensor(
81 inputs_0 = sparse_tensor.SparseTensor(
85 inputs_1 = sparse_tensor.SparseTensor(
90 self.assertIsInstance(output, sparse_tensor.SparseTensor)
[all …]
Dhashing.py26 from tensorflow.python.framework import sparse_tensor
172 if isinstance(inputs, sparse_tensor.SparseTensor):
173 return sparse_tensor.SparseTensor(
213 if isinstance(input_spec, sparse_tensor.SparseTensorSpec):
214 return sparse_tensor.SparseTensorSpec(

1234567891011