| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 57 def _convert_to_sparse_tensor(sp_input): argument
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):
73 return sp_input
91 return [_convert_to_sparse_tensor(sp_input) for sp_input in sp_inputs]
93 return (_convert_to_sparse_tensor(sp_input) for sp_input in sp_inputs)
140 def sparse_expand_dims(sp_input, axis=None, name=None): argument
209 rank = sp_input.dense_shape.get_shape()[0]
211 rank = array_ops.shape(sp_input.dense_shape)[0]
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_reshape_op_test.py | 57 sp_input = sparse_tensor.SparseTensor.from_value(
59 self.assertAllEqual((5, 6), sp_input.get_shape())
60 sp_output = sparse_ops.sparse_reshape(sp_input, shape=(1, 5, 2, 3))
64 sp_input = sparse_tensor.SparseTensor.from_value(
66 self.assertAllEqual((2, 3, 4), sp_input.get_shape())
67 sp_output = sparse_ops.sparse_reshape(sp_input, shape=(2, -1))
72 sp_input = sparse_tensor.SparseTensor.from_value(
75 sparse_ops.sparse_reshape(sp_input, shape=(-1, 2, -1))
79 sp_input = sparse_tensor.SparseTensor.from_value(
82 sparse_ops.sparse_reshape(sp_input, shape=(-1, 7))
[all …]
|
| D | sparse_slice_op_test.py | 86 sp_input = self._SparseTensor_4x6()
87 sp_tensor0 = sparse_ops.sparse_slice(sp_input, [0, 0], [2, 6])
88 sp_tensor1 = sparse_ops.sparse_slice(sp_input, [2, 0], [3, 7])
103 sp_input = self._SparseTensor_5x7()
104 sp_tensor0 = sparse_ops.sparse_slice(sp_input, [0, 0], [5, 3])
105 sp_tensor1 = sparse_ops.sparse_slice(sp_input, [0, 3], [5, 2])
106 sp_tensor2 = sparse_ops.sparse_slice(sp_input, [0, 5], [5, 2])
122 sp_tensor0 = sparse_ops.sparse_slice(sp_input, [0, 0], [5, 2])
123 sp_tensor1 = sparse_ops.sparse_slice(sp_input, [0, 2], [5, 2])
124 sp_tensor2 = sparse_ops.sparse_slice(sp_input, [0, 4], [5, 2])
[all …]
|
| D | sparse_reorder_op_test.py | 53 sp_input = sparse_tensor.SparseTensor.from_value(
55 self.assertAllEqual((5, 6), sp_input.get_shape())
56 sp_output = sparse_ops.sparse_reorder(sp_input)
72 sp_input = self._SparseTensorPlaceholder()
74 sp_output = sparse_ops.sparse_reorder(sp_input)
76 output_val = sess.run(sp_output, {sp_input: input_val})
99 sp_input = self._SparseTensorPlaceholder()
101 sp_output = sparse_ops.sparse_reorder(sp_input)
103 output_val = sess.run(sp_output, {sp_input: input_val})
114 sp_input = sparse_tensor.SparseTensor(input_val.indices,
[all …]
|
| D | sparse_split_op_test.py | 83 sp_input=self._SparseTensor_4x6(), num_split=2, axis=axis))
100 sp_input=self._SparseTensor_5x7(), num_split=3, axis=axis))
116 sp_input=self._SparseTensor_5x7(), num_split=4, axis=axis)
138 sp_input=self._SparseTensor_5x7(), num_split=2, axis=axis))
152 sp_input=self._SparseTensor_5x7(), num_split=3, axis=axis)
170 sp_input=self._SparseTensor_4x6(), num_split=4, axis=axis))
191 sp_input=self._SparseTensor_4x6(), num_split=3, axis=axis))
210 sp_input=self._SparseTensor_4x6(), num_split=6, axis=axis))
232 for sp_input in (self._SparseTensorValue_3x4x2(),
236 sp_input=sp_input, num_split=2, axis=axis)
[all …]
|
| D | sparse_ops_test.py | 77 sp_input = self._SparseTensor_5x6(dtypes.int32)
78 output = sparse_ops.sparse_to_indicator(sp_input, 50)
89 sp_input = self._SparseTensor_5x6(dtypes.int64)
90 output = sparse_ops.sparse_to_indicator(sp_input, 50)
101 sp_input = self._SparseTensor_2x3x4(dtypes.int64)
102 output = sparse_ops.sparse_to_indicator(sp_input, 200)
302 for sp_input in (self._SparseTensorValue_5x6(), self._SparseTensor_5x6()):
304 sp_output = sparse_ops.sparse_retain(sp_input, to_retain)
314 sp_input = self._SparseTensor_5x6()
316 sp_output = sparse_ops.sparse_retain(sp_input, to_retain)
[all …]
|
| D | sparse_serialization_ops_test.py | 73 sp_input = self._SparseTensorValue_5x6(np.arange(6))
74 serialized = serialize_fn(sp_input, out_type=out_type)
79 self.assertAllEqual(indices, sp_input[0])
80 self.assertAllEqual(values, sp_input[1])
81 self.assertAllEqual(shape, sp_input[2])
97 sp_input = self._SparseTensorValue_5x6(np.arange(6))
98 serialized = serialize_fn(sp_input, out_type=out_type)
107 self.assertAllEqual(combined_indices[:6, 1:], sp_input[0])
109 self.assertAllEqual(combined_indices[6:, 1:], sp_input[0])
110 self.assertAllEqual(combined_values[:6], sp_input[1])
[all …]
|
| D | sparse_tensors_map_ops_test.py | 106 sp_input = self._SparseTensorPlaceholder()
109 handle = add_sparse_to_tensors_map(sp_input)
111 handle0_value = sess.run(handle, feed_dict={sp_input: input0_val})
112 handle1_value = sess.run(handle, feed_dict={sp_input: input1_val})
157 sp_input = self._SparseTensorPlaceholder()
160 handle = add_sparse_to_tensors_map(sp_input)
162 handle0_value = sess.run(handle, feed_dict={sp_input: input0_val})
163 handle1_value = sess.run(handle, feed_dict={sp_input: input1_val})
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.sparse.pbtxt | 29 …argspec: "args=[\'sp_input\', \'axis\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'…
37 …argspec: "args=[\'sp_input\', \'default_value\', \'name\'], varargs=None, keywords=None, defaults=…
61 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'output_is_sparse\', \'name\'], varargs=Non…
65 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'output_is_sparse\', \'name\'], varargs=Non…
69 argspec: "args=[\'sp_input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
73 … argspec: "args=[\'sp_input\', \'new_shape\'], varargs=None, keywords=None, defaults=[\'None\'], "
77 …argspec: "args=[\'sp_input\', \'shape\', \'name\'], varargs=None, keywords=None, defaults=[\'None\…
81 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"
97 …argspec: "args=[\'sp_input\', \'start\', \'size\', \'name\'], varargs=None, keywords=None, default…
101 argspec: "args=[\'sp_input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
[all …]
|
| D | tensorflow.io.pbtxt | 145 …argspec: "args=[\'sp_input\', \'out_type\', \'name\'], varargs=None, keywords=None, defaults=[\"<d…
149 …argspec: "args=[\'sp_input\', \'out_type\', \'name\'], varargs=None, keywords=None, defaults=[\"<d…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.sparse.pbtxt | 33 …argspec: "args=[\'sp_input\', \'axis\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'…
41 …argspec: "args=[\'sp_input\', \'default_value\', \'name\'], varargs=None, keywords=None, defaults=…
73 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
77 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
81 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
85 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
89 argspec: "args=[\'sp_input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
93 … argspec: "args=[\'sp_input\', \'new_shape\'], varargs=None, keywords=None, defaults=[\'None\'], "
97 …argspec: "args=[\'sp_input\', \'shape\', \'name\'], varargs=None, keywords=None, defaults=[\'None\…
101 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"
[all …]
|
| D | tensorflow.io.pbtxt | 165 …argspec: "args=[\'sp_input\', \'name\', \'out_type\'], varargs=None, keywords=None, defaults=[\'No…
169 …argspec: "args=[\'sp_input\', \'name\', \'out_type\'], varargs=None, keywords=None, defaults=[\'No…
|
| D | tensorflow.pbtxt | 2081 …argspec: "args=[\'sp_input\', \'name\', \'out_type\'], varargs=None, keywords=None, defaults=[\'No…
2085 …argspec: "args=[\'sp_input\', \'name\', \'out_type\'], varargs=None, keywords=None, defaults=[\'No…
2157 …argspec: "args=[\'sp_input\', \'default_value\', \'name\'], varargs=None, keywords=None, defaults=…
2185 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
2189 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
2193 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
2197 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
2201 argspec: "args=[\'sp_input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
2205 … argspec: "args=[\'sp_input\', \'new_shape\'], varargs=None, keywords=None, defaults=[\'None\'], "
2209 …argspec: "args=[\'sp_input\', \'shape\', \'name\'], varargs=None, keywords=None, defaults=[\'None\…
[all …]
|
| /external/tensorflow/tensorflow/tools/compatibility/testdata/ |
| D | test_file_v1_12.py | 43 sp_input = tf.SparseTensor(
49 serialized_sp = tf.serialize_sparse(sp_input, 'serialize_name', tf.string)
54 sp_input = tf.SparseTensor(
61 sp_input, 'serialize_name', tf.string)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseReduceSumSparse.pbtxt | 40 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
|
| D | api_def_SparseReduceMaxSparse.pbtxt | 40 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
|
| D | api_def_SparseReduceSum.pbtxt | 46 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
|
| D | api_def_SparseReduceMax.pbtxt | 46 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
|
| D | api_def_SparseFillEmptyRows.pbtxt | 59 For example, suppose `sp_input` has shape `[5, 6]` and non-empty values:
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 6396 sparse_ops.sparse_tensor_to_dense(sp_input=st, default_value=-1))
|
| /external/tensorflow/ |
| D | RELEASE.md | 4151 `tf.sparse_split(sp_input, num_split, axis)`. NOTE: we have temporarily
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 14225 For example, suppose `sp_input` has shape `[5, 6]` and non-empty values:
|