Home
last modified time | relevance | path

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

1234567891011

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dsparse_tensor_test.py26 from tensorflow.python.framework import sparse_tensor
43 sp_value = sparse_tensor.SparseTensorValue(indices, values, shape)
45 sparse_tensor.SparseTensor(indices, values, shape),
46 sparse_tensor.SparseTensor.from_value(sp_value),
47 sparse_tensor.SparseTensor.from_value(
48 sparse_tensor.SparseTensor(indices, values, shape))]:
71 tensor = sparse_tensor.SparseTensor(
76 self.assertFalse(sparse_tensor.is_sparse(3))
77 self.assertFalse(sparse_tensor.is_sparse("foo"))
78 self.assertFalse(sparse_tensor.is_sparse(np.array(3)))
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/data/util/
H A Dsparse_test.py28 from tensorflow.python.framework import sparse_tensor
46 (ops.Tensor, sparse_tensor.SparseTensor), True),
48 (sparse_tensor.SparseTensor, sparse_tensor.SparseTensor), True),
49 ("TestCase_6", lambda: (((sparse_tensor.SparseTensor))), True)]
70 lambda: sparse_tensor.SparseTensor,
78 lambda: (sparse_tensor.SparseTensor),
86 (sparse_tensor.SparseTensor, ()), lambda: (tensor_shape.unknown_shape(),
89 ((), sparse_tensor.SparseTensor), lambda: (
97 (sparse_tensor.SparseTensor, (), sparse_tensor.SparseTensor), lambda:
102 ((), sparse_tensor.SparseTensor,
[all …]
H A Dstructure_test.py32 from tensorflow.python.framework import sparse_tensor
58 ("SparseTensor", lambda: sparse_tensor.SparseTensor(
60 lambda: sparse_tensor.SparseTensorSpec, lambda: [dtypes.variant],
77 "b": (sparse_tensor.SparseTensor(
79 sparse_tensor.SparseTensor(
126 ("SparseTensor", lambda: sparse_tensor.SparseTensor(
129 sparse_tensor.SparseTensor(
131 sparse_tensor.SparseTensorValue(
137 sparse_tensor.SparseTensor(
141 sparse_tensor.SparseTensor(
[all …]
H A Dsparse.py19 from tensorflow.python.framework import sparse_tensor
33 return any(c is sparse_tensor.SparseTensor for c in nest.flatten(classes))
49 tensor_shape.unknown_shape() if c is sparse_tensor.SparseTensor else shape
68 dtypes.variant if c is sparse_tensor.SparseTensor else ty
89 if c is sparse_tensor.SparseTensor else tensor
109 sparse_tensor.SparseTensor
110 if isinstance(tensor, sparse_tensor.SparseTensor) else ops.Tensor
127 if sparse_tensor.is_sparse(tensor) else tensor
145 if isinstance(tensor, sparse_tensor.SparseTensor) else tensor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/
H A Dsparse_lower.mlir9 #SV = #sparse_tensor.encoding<{
13 #CSR = #sparse_tensor.encoding<{
17 #DCSR = #sparse_tensor.encoding<{
21 #ST = #sparse_tensor.encoding<{
28 …nsor<10x20xf32, #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ] }>>) outs(%[[OU…
46 …sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ] }>>, tensor<10x20xf32, #sparse_t…
66 …sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ] }>>, tensor<10x20xf32, #sparse_t…
84 …:.*]] = linalg.generic {{{.*}}} ins(%[[ARG0]] : tensor<10x20x30xf64, #sparse_tensor.encoding<{ dim…
87 …T1:.*]] = linalg.generic {{{.*}}} ins(%[[T0]] : tensor<10x20x30xf64, #sparse_tensor.encoding<{ dim…
90 …T2:.*]] = linalg.generic {{{.*}}} ins(%[[T1]] : tensor<10x20x30xf64, #sparse_tensor.encoding<{ dim…
[all …]
H A Dsparse_rewriting.mlir7 #SV= #sparse_tensor.encoding<{ dimLevelType = ["compressed"] }>
9 #CSR = #sparse_tensor.encoding<{
13 #DCSR = #sparse_tensor.encoding<{
23 %1 = sparse_tensor.convert %0 : tensor<100xf64> to tensor<100xf64, #SV>
35 %1 = sparse_tensor.convert %0 : tensor<100xf64> to tensor<100xf64, #SV>
42 …sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ] }>>, tensor<10x10xf64, #sparse_t…
47 %1 = sparse_tensor.convert %0 : tensor<10x10xf64, #CSR> to tensor<10x10xf64, #DCSR>
53 // CHECK: %[[VAL:.*]] = sparse_tensor.convert %[[ARG0]] : tensor<10x10xf64> to tensor<10x10…
54 // CHECK-NEXT: return %[[VAL:.*]] : tensor<10x10xf64, #sparse_tensor.encoding<{ dimLevelType = […
56 %0 = sparse_tensor.convert %arg0 : tensor<10x10xf64> to tensor<10x10xf64, #DCSR>
[all …]
H A Dsparse_transpose.mlir5 #DCSR = #sparse_tensor.encoding<{
26 // CHECK-SAME: %[[A:.*]]: tensor<100x100xf64, #sparse_tensor.encoding<{{{.*}}}>>)
27 // CHECK: return %[[A]] : tensor<100x100xf64, #sparse_tensor.encoding<{{{.*}}}>>
37 // CHECK-SAME: %[[A:.*]]: tensor<100x100xf64, #sparse_tensor.encoding<{{{.*}}}>>)
38 // CHECK: %[[R:.*]] = mhlo.reshape %[[A]] : (tensor<100x100xf64, #sparse_tensor.encoding<{{.*…
50 …= mhlo.reshape %[[A]] : (tensor<100x100xf64>) -> tensor<100x100xf64, #sparse_tensor.encoding<{{.*}…
51 // CHECK: return %[[R]] : tensor<100x100xf64, #sparse_tensor.encoding<{{{.*}}}>>
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dsparse_ops.py29 from tensorflow.python.framework import sparse_tensor
64 if isinstance(sp_input, sparse_tensor.SparseTensorValue):
65 return sparse_tensor.SparseTensor.from_value(sp_input)
66 if not isinstance(sp_input, sparse_tensor.SparseTensor):
131 return sparse_tensor.SparseTensor(indices, values, shape)
236 return sparse_tensor.SparseTensor(
268 return sparse_tensor.SparseTensor(
435 return sparse_tensor.SparseTensor(output_ind, output_val, output_shape)
440 output = sparse_tensor.SparseTensor(output_ind, output_val, output_shape)
568 sparse_classes = (sparse_tensor.SparseTensor, sparse_tensor.SparseTensorValue)
[all …]
H A Dsets_impl.py19 from tensorflow.python.framework import sparse_tensor
48 a = sparse_tensor.convert_to_tensor_or_sparse_tensor(a, name="a")
49 if not isinstance(a, sparse_tensor.SparseTensor):
80 a = sparse_tensor.convert_to_tensor_or_sparse_tensor(a, name="a")
85 b = sparse_tensor.convert_to_tensor_or_sparse_tensor(b, name="b")
88 if (isinstance(a, sparse_tensor.SparseTensor) and
89 not isinstance(b, sparse_tensor.SparseTensor)):
119 if isinstance(a, sparse_tensor.SparseTensor):
120 if isinstance(b, sparse_tensor.SparseTensor):
127 elif isinstance(b, sparse_tensor.SparseTensor):
[all …]
H A Dbincount_ops.py20 from tensorflow.python.framework import sparse_tensor
140 if not isinstance(arr, sparse_tensor.SparseTensor):
143 if not isinstance(weights, sparse_tensor.SparseTensor):
164 if isinstance(arr, sparse_tensor.SparseTensor):
180 if isinstance(arr, sparse_tensor.SparseTensor):
193 if isinstance(arr, sparse_tensor.SparseTensor):
388 if not isinstance(values, sparse_tensor.SparseTensor):
392 if not isinstance(weights, sparse_tensor.SparseTensor):
411 if isinstance(values, sparse_tensor.SparseTensor):
424 if isinstance(values, sparse_tensor.SparseTensor):
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Dfrom_tensor_slices_test.py28 from tensorflow.python.framework import sparse_tensor
95 components = (sparse_tensor.SparseTensorValue(
99 sparse_tensor.SparseTensorValue(
111 (sparse_tensor.SparseTensorValue(
115 sparse_tensor.SparseTensorValue(
119 (sparse_tensor.SparseTensorValue(
123 sparse_tensor.SparseTensorValue(
127 (sparse_tensor.SparseTensorValue(
131 sparse_tensor.SparseTensorValue(
144 sparse_tensor.SparseTensorValue(
[all …]
H A Dfrom_sparse_tensor_slices_test.py25 from tensorflow.python.framework import sparse_tensor
45 get_next = sparse_tensor.SparseTensor(*iterator.get_next())
55 sparse_feed = sparse_tensor.SparseTensorValue(indices, values,
92 sparse_feed = sparse_tensor.SparseTensorValue(
104 get_next = sparse_tensor.SparseTensor(*iterator.get_next())
111 sparse_feed = sparse_tensor.SparseTensorValue(empty_indices, empty_values,
130 sparse_feed = sparse_tensor.SparseTensorValue(empty_indices,
150 sparse_feed = sparse_tensor.SparseTensorValue(empty_indices, empty_values,
174 sparse_components = sparse_tensor.SparseTensor(indices, values, dense_shape)
/aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/
H A Dsequence_feature_column_test.py30 from tensorflow.python.framework import sparse_tensor
170 inputs = sparse_tensor.SparseTensorValue(**inputs_args)
171 expected = sparse_tensor.SparseTensorValue(**expected_args)
244 inputs = sparse_tensor.SparseTensorValue(**inputs_args)
245 expected = sparse_tensor.SparseTensorValue(**expected_args)
295 inputs = sparse_tensor.SparseTensorValue(**inputs_args)
296 expected = sparse_tensor.SparseTensorValue(**expected_args)
313 inputs = sparse_tensor.SparseTensorValue(
315 expected = sparse_tensor.SparseTensorValue(
363 inputs = sparse_tensor.SparseTensorValue(**inputs_args)
[all …]
H A Dfeature_column_test.py43 from tensorflow.python.framework import sparse_tensor
172 builder = _LazyBuilder(features={'a': sparse_tensor.SparseTensor(
326 sparse_tensor.SparseTensor(
515 sparse_tensor.SparseTensor(
709 sparse_tensor.SparseTensorValue(
716 wire_tensor = sparse_tensor.SparseTensor(
737 int_tensor = sparse_tensor.SparseTensor(
741 string_tensor = sparse_tensor.SparseTensor(
745 float_tensor = sparse_tensor.SparseTensor(
762 wire_tensor = sparse_tensor.SparseTensor(
[all …]
H A Dfeature_column_v2_test.py37 from tensorflow.python.framework import sparse_tensor
244 sparse_tensor.SparseTensor(
412 sparse_tensor.SparseTensor(
644 sparse_tensor.SparseTensor(
845 sparse_tensor.SparseTensorValue(
852 wire_tensor = sparse_tensor.SparseTensor(
878 int_tensor = sparse_tensor.SparseTensor(
880 string_tensor = sparse_tensor.SparseTensor(
882 float_tensor = sparse_tensor.SparseTensor(
897 wire_tensor = sparse_tensor.SparseTensor(
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
H A Dsparse_ops_test.py23 from tensorflow.python.framework import sparse_tensor
45 return sparse_tensor.SparseTensor(
55 return sparse_tensor.SparseTensor(
66 return sparse_tensor.SparseTensor(
119 indices = sparse_tensor.SparseTensorValue(
122 values = sparse_tensor.SparseTensorValue(
129 return (sparse_tensor.SparseTensor.from_value(indices),
130 sparse_tensor.SparseTensor.from_value(values))
149 sparse_tensor.SparseTensor.from_value(indices_v)):
151 sparse_tensor.SparseTensor.from_value(values_v)):
[all …]
H A Dsparse_serialization_ops_test.py21 from tensorflow.python.framework import sparse_tensor as sparse_tensor_lib
257 sparse_tensor = self._SparseTensorPlaceholder(dtype=dtypes.string)
258 serialized = serialize_many_fn(sparse_tensor, out_type=out_type)
262 sparse_tensor.indices: indices_value,
263 sparse_tensor.values: values_value,
264 sparse_tensor.dense_shape: shape_value
286 sparse_tensor = self._SparseTensorPlaceholder(dtype=dtypes.string)
287 serialized = serialize_many_fn(sparse_tensor, out_type=out_type)
292 sparse_tensor.indices: indices_value,
293 sparse_tensor.values: values_value,
[all …]
H A Dsparse_cross_op_test.py23 from tensorflow.python.framework import sparse_tensor
58 return sparse_tensor.SparseTensor(
451 return sparse_tensor.SparseTensor(
461 st1 = sparse_tensor.SparseTensor([[0, 0]], [0], [2, 2])
467 st2 = sparse_tensor.SparseTensor([[0, 0]], [0], [2, 2])
473 st3 = sparse_tensor.SparseTensor([[0, 0]], [0], [2, 2])
483 st = sparse_tensor.SparseTensor([[0]], [0], [10]) # 1D SparseTensor
492 st1 = sparse_tensor.SparseTensor([[0, 0]], [0], [10, 10]) # batch size 10
493 st2 = sparse_tensor.SparseTensor([[0, 0]], [0], [7, 10]) # batch size 7
518 out = sparse_tensor.SparseTensor(inds, vals, shapes)
[all …]
H A Dsparse_concat_op_test.py21 from tensorflow.python.framework import sparse_tensor
34 return sparse_tensor.SparseTensor(
49 return sparse_tensor.SparseTensorValue(
54 return sparse_tensor.SparseTensor.from_value(self._SparseTensorValue_3x3())
63 return sparse_tensor.SparseTensorValue(
68 return sparse_tensor.SparseTensor.from_value(self._SparseTensorValue_3x5())
77 return sparse_tensor.SparseTensor(
88 return sparse_tensor.SparseTensor(
102 return sparse_tensor.SparseTensor(
111 return sparse_tensor.SparseTensor(
[all …]
H A Dsparse_reshape_op_test.py23 from tensorflow.python.framework import sparse_tensor
33 return sparse_tensor.SparseTensor(
43 return sparse_tensor.SparseTensorValue(ind, val, shape)
50 return sparse_tensor.SparseTensorValue(ind, val, shape)
53 sp_input = sparse_tensor.SparseTensor.from_value(
60 sp_input = sparse_tensor.SparseTensor.from_value(
68 sp_input = sparse_tensor.SparseTensor.from_value(
75 sp_input = sparse_tensor.SparseTensor.from_value(
82 sp_input = sparse_tensor.SparseTensor.from_value(
269 sp_input = sparse_tensor.SparseTensor.from_value(input_val)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_from_sparse_op_test.py20 from tensorflow.python.framework import sparse_tensor
31 st = sparse_tensor.SparseTensor(
40 st = sparse_tensor.SparseTensor(
49 st1 = sparse_tensor.SparseTensor(indices=[[0]], values=[0], dense_shape=[3])
53 st2 = sparse_tensor.SparseTensor(
59 st3 = sparse_tensor.SparseTensor(
68 st1 = sparse_tensor.SparseTensor(
72 st2 = sparse_tensor.SparseTensor(
88 st1 = sparse_tensor.SparseTensor(
94 st2 = sparse_tensor.SparseTensor(
[all …]
/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/
H A Dutils.py8 def sparse_tensor_to_rpc_format(sparse_tensor): argument
15 sparse_tensor = sparse_tensor.coalesce()
16 return [sparse_tensor.indices(), sparse_tensor.values(), sparse_tensor.size()]
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dinput_test.py26 from tensorflow.python.framework import sparse_tensor
453 sparse_counter = sparse_tensor.SparseTensor(
566 sparse_counter = sparse_tensor.SparseTensor(
602 sparse_counter = sparse_tensor.SparseTensor(
642 sparse_counter = sparse_tensor.SparseTensor(
700 sparse_counter = sparse_tensor.SparseTensor(
771 sparse = sparse_tensor.SparseTensor(
779 sparse = sparse_tensor.SparseTensor(
787 sparse = sparse_tensor.SparseTensor(
797 sparse = sparse_tensor.SparseTensor(
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/chlo/
H A Dsparse_chlo_legalize_to_linalg.mlir3 #CSR = #sparse_tensor.encoding<{
23 // CHECK-SAME: #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ] }>>
25 // CHECK: sparse_tensor.unary %{{.*}} : f32 to f32
30 // CHECK: sparse_tensor.yield %{{.*}} : f32
50 // CHECK: %[[TMP_2:.*]] = sparse_tensor.unary %[[TMP_arg1]] : f32 to f32
56 // CHECK: sparse_tensor.yield %[[TMP_5]] : f32
77 // CHECK: %[[TMP_2:.*]] = sparse_tensor.unary %[[TMP_arg1]] : f32 to f32
83 // CHECK: sparse_tensor.yield %[[TMP_5]] : f32
/aosp_15_r20/external/tensorflow/tensorflow/core/util/sparse/
H A DBUILD20 "sparse_tensor.cc",
21 "sparse_tensor.h",
30 "sparse_tensor.h",
40 "sparse_tensor.h",
49 "sparse_tensor.cc",

1234567891011