| /external/tensorflow/tensorflow/python/data/util/ |
| sparse_test.py | 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),
136 "classes": (sparse_tensor.SparseTensor, (),
[all...] |
| sparse.py | 37 return any(c is sparse_tensor.SparseTensor for c in nest.flatten(classes))
50 `tf.SparseTensor` and matching contents of `shapes` otherwise
53 tensor_shape.unknown_shape() if c is sparse_tensor.SparseTensor else shape
68 `dtypes.variant` at positions where `classes` contains `tf.SparseTensor` and
72 dtypes.variant if c is sparse_tensor.SparseTensor else ty
93 if c is sparse_tensor.SparseTensor else tensor
109 `tf.SparseTensor` at positions where `tensors` contains a sparse tensor and
113 sparse_tensor.SparseTensor
114 if isinstance(tensor, sparse_tensor.SparseTensor) else ops.Tensor
149 if isinstance(tensor, sparse_tensor.SparseTensor) else tenso
[all...] |
| structure_test.py | 47 (lambda: sparse_tensor.SparseTensor(
58 "b": (sparse_tensor.SparseTensor(
60 sparse_tensor.SparseTensor(
82 (lambda: sparse_tensor.SparseTensor(
85 sparse_tensor.SparseTensor(
93 sparse_tensor.SparseTensor(
97 sparse_tensor.SparseTensor(
116 sparse_tensor.SparseTensor(
138 (lambda: sparse_tensor.SparseTensor(
143 "b": (sparse_tensor.SparseTensor(
[all...] |
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| ragged_from_sparse_op_test.py | 36 st = sparse_tensor.SparseTensor(
45 st = sparse_tensor.SparseTensor(
54 st1 = sparse_tensor.SparseTensor(indices=[[0]], values=[0], dense_shape=[3])
58 st2 = sparse_tensor.SparseTensor(
64 st3 = sparse_tensor.SparseTensor(
73 st1 = sparse_tensor.SparseTensor(
77 st2 = sparse_tensor.SparseTensor(
93 st1 = sparse_tensor.SparseTensor(
96 r'.*SparseTensor is not right-ragged'):
99 st2 = sparse_tensor.SparseTensor(
[all...] |
| /external/tensorflow/tensorflow/core/util/sparse/ |
| sparse_tensor_test.cc | 97 SparseTensor result;
99 EXPECT_EQ(SparseTensor::Create(ix, vals, TensorShape({10, 10, 10}), {0, 1, 2},
110 SparseTensor result;
112 EXPECT_EQ(SparseTensor::Create(ix, vals, TensorShape({10, 10, 10}), {0, 1, 2},
123 SparseTensor result;
125 EXPECT_EQ(SparseTensor::Create(ix, vals, TensorShape({10, 10, 10}), {0, 1, 2},
136 SparseTensor result;
138 EXPECT_EQ(SparseTensor::Create(ix, vals, TensorShape({10, 10, 10}), {0, 1, 2},
149 SparseTensor result;
152 SparseTensor::Create(ix, vals, TensorShape({10, 10, 10}), {0, 1}, &result
[all...] |
| sparse_tensor.h | 42 class SparseTensor {
48 const VarDimArray order, SparseTensor* result) {
70 "Order length must be SparseTensor rank.");
74 "Shape rank must be SparseTensor rank.");
77 *result = SparseTensor(ix, vals, shape, order);
82 SparseTensor* result) {
88 SparseTensor* result) {
93 const VarDimArray order, SparseTensor* result) {
97 SparseTensor() : dims_(0) {}
100 SparseTensor(Tensor ix, Tensor vals, const TensorShape& shape
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_reorder_op.cc | 63 sparse::SparseTensor input_sp;
65 context, sparse::SparseTensor::Create(input_ind, input_val, input_shape,
73 sparse::SparseTensor reordered_sp;
75 sparse::SparseTensor::Create(tensor::DeepCopy(input_ind),
|
| sparse_slice_op.cc | 69 sparse::SparseTensor sparse_tensor;
71 sparse::SparseTensor::Create(
80 const sparse::SparseTensor output =
81 sparse::SparseTensor::Slice<T>(sparse_tensor, start, size);
|
| sparse_split_op.cc | 66 sparse::SparseTensor sparse_tensor;
68 sparse::SparseTensor::Create(
72 std::vector<sparse::SparseTensor> outputs;
74 sparse::SparseTensor::Split<T>(sparse_tensor, split_dim,
|
| sparse_tensors_map_ops.cc | 40 using sparse::SparseTensor;
54 Status AddSparseTensor(OpKernelContext* ctx, const SparseTensor& sp,
81 std::vector<SparseTensor>* sparse_tensors) {
91 "Unable to find SparseTensor: ", handle, " in map: ", name_);
96 SparseTensor tensor;
97 TF_RETURN_IF_ERROR(SparseTensor::Create(*ix, *values, shape, &tensor));
199 SparseTensor st;
200 OP_REQUIRES_OK(context, SparseTensor::Create(*input_indices, *input_values,
254 "Rank of input SparseTensor should be > 1, but saw rank: ", rank));
259 SparseTensor input_st
[all...] |
| deserialize_sparse_string_op.cc | 40 using sparse::SparseTensor;
143 "SparseTensor[",
144 i, "] was: ", shape.dims() - 1, " but rank of SparseTensor[", i,
165 std::vector<SparseTensor> tensors;
168 SparseTensor tensor;
169 OP_REQUIRES_OK(context, SparseTensor::Create(indices[i], values[i], shape,
174 gtl::optional<SparseTensor> maybe_output;
177 maybe_output = SparseTensor::Concat<T>(tensors); \
191 SparseTensor& output = maybe_output.value();
251 "Requested SparseTensor of type ", DataTypeString(values_dtype)
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| sparse_tensor_test.py | 40 sparse_tensor.SparseTensor(indices, values, shape),
41 sparse_tensor.SparseTensor.from_value(sp_value),
42 sparse_tensor.SparseTensor.from_value(
43 sparse_tensor.SparseTensor(indices, values, shape))]:
64 sparse_tensor.is_sparse(sparse_tensor.SparseTensor([[0]], [0], [1])))
71 sp = sparse_tensor.SparseTensor([[0, 0], [1, 2]], [1.0, 3.0], [3, 4])
100 st = sparse_tensor.SparseTensor.from_value(sparse_tensor_value)
|
| composite_tensor_test.py | 66 st1 = sparse_tensor.SparseTensor([[0, 3], [7, 2]], [1, 2], [10, 10])
67 st2 = sparse_tensor.SparseTensor([[1, 2, 3]], ['a'], [10, 10, 10])
80 st1 = sparse_tensor.SparseTensor([[0, 3], [7, 2]], [1, 2], [10, 10])
81 st2 = sparse_tensor.SparseTensor([[1, 2, 3]], ['a'], [10, 10, 10])
90 st1 = sparse_tensor.SparseTensor([[0]], [0], [100])
91 st2 = sparse_tensor.SparseTensor([[0, 3]], ['x'], [100, 100])
|
| sparse_tensor.py | 38 @tf_export("sparse.SparseTensor", "SparseTensor")
39 class SparseTensor(_TensorLike, composite_tensor.CompositeTensor):
44 collected into a `SparseTensor` class for ease of use. If you have separate
45 `indices`, `values`, and `dense_shape` tensors, wrap them in a `SparseTensor`
48 Concretely, the sparse tensor `SparseTensor(indices, values, dense_shape)`
50 of values and number of dimensions in the `SparseTensor`, respectively:
80 `SparseTensor` objects are constructed, but most ops assume correct ordering.
87 SparseTensor(indices=[[0, 0], [1, 2]], values=[1, 2], dense_shape=[3, 4])
102 raise TypeError("Neither a SparseTensor nor SparseTensorValue: %s."
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| sets_impl.py | 39 a: `SparseTensor`, with indices sorted in row-major order.
52 if not isinstance(a, sparse_tensor.SparseTensor):
53 raise TypeError("Expected `SparseTensor`, got %s." % a)
72 a: `Tensor` or `SparseTensor` of the same type as `b`.
73 b: `Tensor` or `SparseTensor` of the same type as `a`.
77 `Tensor` or `SparseTensor`, and `flipped` indicates whether the order has
87 if (isinstance(a, sparse_tensor.SparseTensor) and
88 not isinstance(b, sparse_tensor.SparseTensor)):
99 a: `Tensor` or `SparseTensor` of the same type as `b`. If sparse, indices
101 b: `Tensor` or `SparseTensor` of the same type as `a`. Must b
[all...] |
| sparse_ops.py | 19 See also `tf.SparseTensor`.
54 """Convert `sp_input` to `SparseTensor` and return it.
57 sp_input: `SparseTensor` or `SparseTensorValue`.
60 `sp_input` converted to `SparseTensor`.
63 ValueError: if `sp_input` is neither `SparseTensor` nor `SparseTensorValue`.
66 return sparse_tensor.SparseTensor.from_value(sp_input)
67 if not isinstance(sp_input, sparse_tensor.SparseTensor):
68 raise TypeError("Input must be a SparseTensor.")
73 """Convert `sp_inputs` to `SparseTensor` objects and return them.
76 sp_inputs: `list` or `tuple` of `SparseTensor` or `SparseTensorValue
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| batch_features.cc | 99 auto order_dims = sparse::SparseTensor::VarDimArray({0, 1});
100 sparse::SparseTensor sparse_tensor;
101 TF_RETURN_IF_ERROR(sparse::SparseTensor::Create(
141 auto order_dims = sparse::SparseTensor::VarDimArray({0, 1});
142 sparse::SparseTensor sparse_tensor;
143 TF_RETURN_IF_ERROR(sparse::SparseTensor::Create(
|
| batch_features.h | 85 std::vector<sparse::SparseTensor> sparse_float_feature_columns_;
88 std::vector<sparse::SparseTensor> sparse_int_feature_columns_;
|
| examples_iterable_test.cc | 47 sparse::SparseTensor sparse_float_tensor1;
49 sparse::SparseTensor::Create(sparse_float_indices1, sparse_float_values1,
56 sparse::SparseTensor sparse_float_tensor2;
58 sparse::SparseTensor::Create(sparse_float_indices2, sparse_float_values2,
64 sparse::SparseTensor sparse_int_tensor1;
66 sparse::SparseTensor::Create(sparse_int_indices1, sparse_int_values1,
72 sparse::SparseTensor sparse_int_tensor2;
74 sparse::SparseTensor::Create(sparse_int_indices2, sparse_int_values2,
|
| examples_iterable.cc | 25 const std::vector<sparse::SparseTensor>& sparse_float_feature_columns,
26 const std::vector<sparse::SparseTensor>& sparse_int_feature_columns,
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| feature_column_ops_test.py | 58 rating_tensor = sparse_tensor.SparseTensor(
76 rating_tensor = sparse_tensor.SparseTensor(
131 wire_tensor = sparse_tensor.SparseTensor(
154 wire_tensor = sparse_tensor.SparseTensor(
181 # While the input is a dense Tensor, the output should be a SparseTensor.
182 self.assertIsInstance(output, sparse_tensor.SparseTensor)
190 wire_tensor = sparse_tensor.SparseTensor(
216 wire_tensor = sparse_tensor.SparseTensor(
246 # While the input is a dense Tensor, the output should be a SparseTensor.
247 self.assertIsInstance(output, sparse_tensor.SparseTensor)
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/ops/ |
| sparse_feature_cross_op.py | 47 """Crosses a list of Tensor or SparseTensor objects.
52 inputs: List of `SparseTensor` or `Tensor` to be crossed.
64 A `SparseTensor` with the crossed features.
68 TypeError: If the inputs aren't either SparseTensor or Tensor.
72 if not all(isinstance(i, sparse_tensor.SparseTensor) or
77 if isinstance(i, sparse_tensor.SparseTensor)]
79 if not isinstance(i, sparse_tensor.SparseTensor)]
122 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
|
| /external/tensorflow/tensorflow/contrib/libsvm/python/ops/ |
| libsvm_ops.py | 42 features: A `SparseTensor` of the shape `[input_shape, num_features]`.
47 return sparse_tensor.SparseTensor(indices, values, shape), labels
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| prettyprint_ops_test.py | 41 a = sparse_tensor.SparseTensor(
43 b = sparse_tensor.SparseTensor(
|
| prettyprint_ops.py | 48 elif isinstance(t, sparse_tensor.SparseTensor):
49 t_type_str = "Type: SparseTensor ({})".format(t.dtype.name)
55 raise ValueError("t must be a Tensor, SparseTensor, TensorArray or "
61 if isinstance(t, sparse_tensor.SparseTensor):
76 if isinstance(t, sparse_tensor.SparseTensor):
108 input_: A Tensor/SparseTensor/TensorArray to print when it is evaluated.
125 ValueError: If the tensor `input_` is not a Tensor, SparseTensor or
133 "Tensor, SparseTensor or TensorArray")
147 elif isinstance(input_, sparse_tensor.SparseTensor):
153 input_ = sparse_tensor.SparseTensor(
[all...] |
