| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/ |
| data_ops.py | 127 is_sparse = isinstance(data[k], sparse_tensor.SparseTensor)
139 sparse_tensor.SparseTensor(data[
165 elif isinstance(data, sparse_tensor.SparseTensor):
203 labels, sparse_tensor.SparseTensor) else labels[k]
209 if isinstance(labels, sparse_tensor.SparseTensor):
|
| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_concat_op.cc | 124 std::vector<sparse::SparseTensor> sp_inputs;
127 sparse::SparseTensor tensor;
129 sparse::SparseTensor::Create(
136 sparse::SparseTensor concat = sparse::SparseTensor::Concat<T>(sp_inputs);
|
| sparse_softmax_op.cc | 31 using tensorflow::sparse::SparseTensor;
72 SparseTensor st;
74 context, SparseTensor::Create(
97 // The SparseTensor has logical shape [..., b, c], where the
|
| sparse_reduce_op.cc | 30 using tensorflow::sparse::SparseTensor;
46 // Compute common reduce parameters that'll be used for SparseTensor
54 ReduceDetails SparseTensorReduceHelper(const SparseTensor &sp,
107 // indices and values are validated in SparseTensor ctor.
175 SparseTensor sp;
176 OP_REQUIRES_OK(ctx, SparseTensor::Create(
265 SparseTensor sp;
266 OP_REQUIRES_OK(ctx, SparseTensor::Create(tensor::DeepCopy(*indices_t),
273 // Count nnzs in the output SparseTensor.
|
| sparse_to_dense_op.cc | 119 // Assume SparseTensor is lexicographically sorted.
122 sparse::SparseTensor st;
124 sparse::SparseTensor::Create(indices_shaped, sparse_values_b,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| sparse_ops_test.py | 40 # SparseTensor gen funcs.
49 return sparse_tensor.SparseTensor(
59 return sparse_tensor.SparseTensor(
70 return sparse_tensor.SparseTensor(
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)):
214 ids = sparse_tensor.SparseTensor(
218 values = sparse_tensor.SparseTensor(
[all...] |
| sparse_concat_op_test.py | 38 return sparse_tensor.SparseTensor(
58 return sparse_tensor.SparseTensor.from_value(self._SparseTensorValue_3x3())
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(
130 return sparse_tensor.SparseTensor(
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| sdca_estimator_test.py | 123 sparse_tensor.SparseTensor(
151 sparse_tensor.SparseTensor(
156 sparse_tensor.SparseTensor(
182 sparse_tensor.SparseTensor(
188 sparse_tensor.SparseTensor(
214 sparse_tensor.SparseTensor(
219 sparse_tensor.SparseTensor(
250 sparse_tensor.SparseTensor(
289 sparse_tensor.SparseTensor(
367 sparse_tensor.SparseTensor(
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| encoders.py | 44 ids: `[batch_size, doc_length]` `Tensor` or `SparseTensor` of type
48 sparse_lookup: `bool`, if `True`, converts ids to a `SparseTensor`
77 if isinstance(ids, sparse_tensor.SparseTensor):
84 if isinstance(ids, sparse_tensor.SparseTensor):
|
| embedding_ops_test.py | 77 sparse_ids = sparse_tensor_lib.SparseTensor(
82 sparse_weights = sparse_tensor_lib.SparseTensor(
103 sparse_ids = sparse_tensor_lib.SparseTensor(
108 sparse_weights = sparse_tensor_lib.SparseTensor(
346 sparse_tensor = sparse_tensor_lib.SparseTensor(
484 sp_values = sparse_tensor_lib.SparseTensor(
498 sp_values = sparse_tensor_lib.SparseTensor(
512 sp_values = sparse_tensor_lib.SparseTensor(
537 sp_values = sparse_tensor_lib.SparseTensor(
555 sp_values_a = sparse_tensor_lib.SparseTensor(
[all...] |
| /external/tensorflow/tensorflow/tools/compatibility/testdata/ |
| test_file_v1_12.py | 36 sp_input = tf.SparseTensor(
47 sp_input = tf.SparseTensor(
|
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| batch_benchmark.py | 36 tensor = sparse_tensor.SparseTensor(
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| tf_utils_test.py | 41 sparse_tensor.SparseTensor(
49 sparse_tensor.SparseTensor(
70 sparse_tensor.SparseTensor(
79 sparse_tensor.SparseTensor(
|
| composite_tensor_support_test.py | 49 elif isinstance(inputs, sparse_tensor.SparseTensor):
78 return sparse_tensor.SparseTensor(indices, values, dense_shape=shape)
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| tfexample_decoder.py | 140 if isinstance(side, sparse_tensor.SparseTensor):
197 if isinstance(shape_dim, sparse_tensor.SparseTensor):
201 if isinstance(tensor, sparse_tensor.SparseTensor):
281 class SparseTensor(ItemHandler):
300 shape: the output shape of the SparseTensor. If `shape_key` is not
302 densify: whether to convert the SparseTensor into a dense Tensor.
304 specified in the `SparseTensor`.
317 super(SparseTensor, self).__init__(keys)
324 if isinstance(shape, sparse_tensor.SparseTensor):
338 tensor = sparse_tensor.SparseTensor(new_indices, values.values, shape
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| sparse_grad.py | 55 sp_unordered = sparse_tensor.SparseTensor(
70 as `SparseTensor` objects. This op takes in the upstream gradient w.r.t.
124 the sliced `SparseTensor`, and outputs the gradients w.r.t.
125 the non-empty values of input `SparseTensor`.
238 sparse_tensor.SparseTensor(scaled_indices, dy_val, y_shape))
275 sp_output = sparse_tensor.SparseTensor(indices, out_vals, shape)
276 sp_grad = sparse_tensor.SparseTensor(indices, grad, shape)
277 sp_product = sparse_tensor.SparseTensor(
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| linear_test.py | 88 sparse_tensor.SparseTensor(
110 sparse_tensor.SparseTensor(
113 sparse_tensor.SparseTensor(
185 sparse_tensor.SparseTensor(
249 sparse_tensor.SparseTensor(
473 sparse_tensor.SparseTensor(
516 sparse_tensor.SparseTensor(
631 sparse_tensor.SparseTensor(
653 sparse_tensor.SparseTensor(
674 sparse_tensor.SparseTensor(
[all...] |
| dnn_linear_combined_test.py | 129 sparse_tensor.SparseTensor(
165 sparse_tensor.SparseTensor(
170 sparse_tensor.SparseTensor(
264 sparse_tensor.SparseTensor(
317 sparse_tensor.SparseTensor(
390 # The following shows how to provide the SparseTensor data for
392 features['dummy_sparse_column'] = sparse_tensor.SparseTensor(
439 # The following shows how to provide the SparseTensor data for
441 features['dummy_sparse_column'] = sparse_tensor.SparseTensor(
476 sparse_tensor.SparseTensor(
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| input_test.py | 471 sparse_counter = sparse_tensor.SparseTensor(
588 sparse_counter = sparse_tensor.SparseTensor(
625 sparse_counter = sparse_tensor.SparseTensor(
666 sparse_counter = sparse_tensor.SparseTensor(
725 sparse_counter = sparse_tensor.SparseTensor(
798 sparse = sparse_tensor.SparseTensor(
806 sparse = sparse_tensor.SparseTensor(
814 sparse = sparse_tensor.SparseTensor(
824 sparse = sparse_tensor.SparseTensor(
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| examples_iterable.h | 37 const std::vector<sparse::SparseTensor>& sparse_float_feature_columns,
38 const std::vector<sparse::SparseTensor>& sparse_int_feature_columns,
|
| /external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/ |
| masked_matmul_ops_test.py | 74 result = sparse_tensor.SparseTensor(
76 true_result = sparse_tensor.SparseTensor(
|
| wals_solver_ops_test.py | 33 return sparse_tensor.SparseTensor(ind, val, shape)
|
| /external/tensorflow/tensorflow/contrib/layers/python/ops/ |
| sparse_ops.py | 15 """Ops to work with `SparseTensor`."""
56 """Converts dense `Tensor` to `SparseTensor`, dropping `ignore_value` cells.
61 absent from the return `SparseTensor`. If `None`, default value of
65 A `SparseTensor` with the same shape as `dense_tensor`.
75 return sparse_tensor.SparseTensor(
113 `sparse_ids` in `SparseTensor` format:
127 absent from the returned `SparseTensor`. If `None`, default value of
132 `SparseTensor` of type `dtype` and shape `(d0, ..., dn, max_num_labels)`,
135 shape is (2, 2, 3)). The values of this `SparseTensor` are in the range
147 # SparseTensor requires int64 indices
[all...] |
| /external/tensorflow/tensorflow/python/saved_model/ |
| utils_impl.py | 51 tensor: Tensor or SparseTensor whose name, dtype and shape are used to
71 if isinstance(tensor, sparse_tensor.SparseTensor):
123 """Returns the Tensor or SparseTensor described by a TensorInfo proto.
126 tensor_info: A TensorInfo proto describing a Tensor or SparseTensor.
133 The Tensor or SparseTensor in `graph` described by `tensor_info`.
147 return sparse_tensor.SparseTensor(
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| sparse_tensor_slice_dataset_op.cc | 34 const sparse::SparseTensor& sparse_tensor)
136 // `SparseTensor`, and we have already read the value from the
144 // The current position is empty in the input `SparseTensor`,
212 const sparse::SparseTensor sparse_tensor_;
251 // that batches elements into rows of a SparseTensor).
257 errors::Unimplemented("The SparseTensor must be ordered in the batch "
263 sparse::SparseTensor tensor;
265 ctx, sparse::SparseTensor::Create(
|
