Home
last modified time | relevance | path

Searched refs:sp_input (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py53 def _convert_to_sparse_tensor(sp_input): argument
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):
69 return sp_input
87 return [_convert_to_sparse_tensor(sp_input) for sp_input in sp_inputs]
89 return (_convert_to_sparse_tensor(sp_input) for sp_input in sp_inputs)
104 def sparse_expand_dims(sp_input, axis=None, name=None): argument
123 rank = sp_input.dense_shape.get_shape()[0]
126 with ops.name_scope(name, default_name="expand_dims", values=[sp_input]):
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_reshape_op_test.py55 sp_input = sparse_tensor.SparseTensor.from_value(
57 self.assertAllEqual((5, 6), sp_input.get_shape())
58 sp_output = sparse_ops.sparse_reshape(sp_input, shape=(1, 5, 2, 3))
62 sp_input = sparse_tensor.SparseTensor.from_value(
64 self.assertAllEqual((2, 3, 4), sp_input.get_shape())
65 sp_output = sparse_ops.sparse_reshape(sp_input, shape=(2, -1))
70 sp_input = sparse_tensor.SparseTensor.from_value(
73 sparse_ops.sparse_reshape(sp_input, shape=(-1, 2, -1))
77 sp_input = sparse_tensor.SparseTensor.from_value(
80 sparse_ops.sparse_reshape(sp_input, shape=(-1, 7))
[all …]
Dsparse_slice_op_test.py86 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])
104 sp_input = self._SparseTensor_5x7()
105 sp_tensor0 = sparse_ops.sparse_slice(sp_input, [0, 0], [5, 3])
106 sp_tensor1 = sparse_ops.sparse_slice(sp_input, [0, 3], [5, 2])
107 sp_tensor2 = sparse_ops.sparse_slice(sp_input, [0, 5], [5, 2])
123 sp_tensor0 = sparse_ops.sparse_slice(sp_input, [0, 0], [5, 2])
124 sp_tensor1 = sparse_ops.sparse_slice(sp_input, [0, 2], [5, 2])
125 sp_tensor2 = sparse_ops.sparse_slice(sp_input, [0, 4], [5, 2])
[all …]
Dsparse_reorder_op_test.py53 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 …]
Dsparse_ops_test.py77 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 …]
Dsparse_serialization_ops_test.py73 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 …]
Dsparse_split_op_test.py83 sp_input=self._SparseTensor_4x6(), num_split=2, axis=0)
101 sp_input=self._SparseTensor_5x7(), num_split=3, axis=1)
119 sp_input=self._SparseTensor_5x7(), num_split=4, axis=1)
141 sp_input=self._SparseTensor_5x7(), num_split=2, axis=0)
156 sp_input=self._SparseTensor_5x7(), num_split=3, axis=0)
178 sp_input=self._SparseTensor_4x6(), num_split=4, axis=0)
201 sp_input=self._SparseTensor_4x6(), num_split=3, axis=1)
220 sp_input=self._SparseTensor_4x6(), num_split=6, axis=1)
246 for sp_input in (self._SparseTensorValue_3x4x2(),
250 sp_input=sp_input, num_split=2, axis=1)
[all …]
Dsparse_tensors_map_ops_test.py106 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/v1/
Dtensorflow.sparse.pbtxt29 …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=…
65 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
69 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
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\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
85 … argspec: "args=[\'sp_input\', \'new_shape\'], varargs=None, keywords=None, defaults=[\'None\'], "
89 …argspec: "args=[\'sp_input\', \'shape\', \'name\'], varargs=None, keywords=None, defaults=[\'None\…
93 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"
[all …]
Dtensorflow.io.pbtxt149 …argspec: "args=[\'sp_input\', \'name\', \'out_type\'], varargs=None, keywords=None, defaults=[\'No…
153 …argspec: "args=[\'sp_input\', \'name\', \'out_type\'], varargs=None, keywords=None, defaults=[\'No…
Dtensorflow.pbtxt1997 …argspec: "args=[\'sp_input\', \'name\', \'out_type\'], varargs=None, keywords=None, defaults=[\'No…
2001 …argspec: "args=[\'sp_input\', \'name\', \'out_type\'], varargs=None, keywords=None, defaults=[\'No…
2073 …argspec: "args=[\'sp_input\', \'default_value\', \'name\'], varargs=None, keywords=None, defaults=…
2101 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
2105 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
2109 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
2113 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
2117 argspec: "args=[\'sp_input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
2121 … argspec: "args=[\'sp_input\', \'new_shape\'], varargs=None, keywords=None, defaults=[\'None\'], "
2125 …argspec: "args=[\'sp_input\', \'shape\', \'name\'], varargs=None, keywords=None, defaults=[\'None\…
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.sparse.pbtxt29 …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=…
53 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'output_is_sparse\', \'name\'], varargs=Non…
57 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'output_is_sparse\', \'name\'], varargs=Non…
61 argspec: "args=[\'sp_input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
65 … argspec: "args=[\'sp_input\', \'new_shape\'], varargs=None, keywords=None, defaults=[\'None\'], "
69 …argspec: "args=[\'sp_input\', \'shape\', \'name\'], varargs=None, keywords=None, defaults=[\'None\…
73 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"
89 …argspec: "args=[\'sp_input\', \'start\', \'size\', \'name\'], varargs=None, keywords=None, default…
93 argspec: "args=[\'sp_input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
[all …]
Dtensorflow.io.pbtxt133 …argspec: "args=[\'sp_input\', \'out_type\', \'name\'], varargs=None, keywords=None, defaults=[\"<d…
137 …argspec: "args=[\'sp_input\', \'out_type\', \'name\'], varargs=None, keywords=None, defaults=[\"<d…
/external/tensorflow/tensorflow/contrib/layers/python/ops/
Dsparse_feature_cross_op.py81 indices = [sp_input.indices for sp_input in sparse_inputs]
82 values = [sp_input.values for sp_input in sparse_inputs]
83 shapes = [sp_input.dense_shape for sp_input in sparse_inputs]
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dfactorization_ops_test.py43 def count_rows(self, sp_input): argument
45 array_ops.shape(array_ops.unique(sp_input.indices[:, 0])[0])[0],
48 def count_cols(self, sp_input): argument
50 array_ops.shape(array_ops.unique(sp_input.indices[:, 1])[0])[0],
146 sp_input=sparse_tensor.SparseTensor(
201 sp_input=sp_feeder, transpose_input=False)
217 sp_input=sp_feeder,
223 sp_input=sp_feeder, transpose_input=False)
273 sp_input=sp_feeder, transpose_input=False)
290 sp_input=sp_feeder,
[all …]
Dfactorization_ops.py676 def update_row_factors(self, sp_input=None, transpose_input=False): argument
707 True, sp_input=sp_input, transpose_input=transpose_input)
709 def update_col_factors(self, sp_input=None, transpose_input=False): argument
740 False, sp_input=sp_input, transpose_input=transpose_input)
743 sp_input=None, argument
771 sp_input=sp_input,
776 sp_input=None, argument
804 sp_input=sp_input,
810 sp_input=None, argument
844 assert isinstance(sp_input, sparse_tensor.SparseTensor)
[all …]
Dwals.py230 def create_axis_ops(sp_input, num_items, update_fn, axis_name): argument
258 _, update_op, loss, reg, sum_weights = update_fn(sp_input)
259 input_indices = sp_input.indices[:, 0]
283 lambda x: model.update_row_factors(sp_input=x, transpose_input=False),
288 lambda x: model.update_col_factors(sp_input=x, transpose_input=True),
324 sp_input=input_rows,
330 sp_input=input_cols,
353 sp_input=input_rows, transpose_input=False)
357 sp_input=input_cols, transpose_input=True)
/external/tensorflow/tensorflow/tools/compatibility/testdata/
Dtest_file_v1_12.py36 sp_input = tf.SparseTensor(
42 serialized_sp = tf.serialize_sparse(sp_input, 'serialize_name', tf.string)
47 sp_input = tf.SparseTensor(
54 sp_input, 'serialize_name', tf.string)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseReduceSumSparse.pbtxt40 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
Dapi_def_SparseReduceMaxSparse.pbtxt40 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
Dapi_def_SparseReduceMax.pbtxt46 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
Dapi_def_SparseReduceSum.pbtxt46 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
Dapi_def_SparseFillEmptyRows.pbtxt59 For example, suppose `sp_input` has shape `[5, 6]` and non-empty values:
/external/tensorflow/
DRELEASE.md1478 `tf.sparse_split(sp_input, num_split, axis)`. NOTE: we have temporarily