| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| map_defun_op_test.py | 280 st = sparse_tensor.SparseTensor(
297 st = sparse_tensor.SparseTensor(
322 st = sparse_tensor.SparseTensor(
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| map_vectorization_test.py | 280 y for y in parse_result if not isinstance(y, sparse_tensor.SparseTensor)
529 tensor = sparse_tensor.SparseTensor(
549 return sparse_tensor.SparseTensor(
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| optional_test.py | 257 ("SparseTensor", lambda: sparse_tensor.SparseTensor(
310 ("SparseTensor", sparse_tensor.SparseTensorValue(
313 lambda: sparse_tensor.SparseTensor(
322 "b": sparse_tensor.SparseTensor(
|
| dataset_test.py | 100 sparse_tensor.SparseTensor(
217 ("SparseTensor", lambda: sparse_tensor.SparseTensor(
|
| /external/tensorflow/tensorflow/python/ops/ |
| embedding_ops.py | 393 sp_ids: N x M `SparseTensor` of int64 ids where N is typically batch size
395 sp_weights: either a `SparseTensor` of float / double weights, or `None` to
447 TypeError: If `sp_ids` is not a `SparseTensor`, or if `sp_weights` is
448 neither `None` nor `SparseTensor`.
461 if not isinstance(sp_ids, sparse_tensor.SparseTensor):
462 raise TypeError("sp_ids must be SparseTensor")
465 if not isinstance(sp_weights, sparse_tensor.SparseTensor):
466 raise TypeError("sp_weights must be either None or SparseTensor")
567 sp_ids: N x M `SparseTensor` of int64 ids where N is typically batch size
569 sp_weights: either a `SparseTensor` of float / double weights, or `None` t
[all...] |
| string_ops.py | 200 """Split elements of `source` based on `delimiter` into a `SparseTensor`.
203 element of `source` based on `delimiter` and return a `SparseTensor`
234 A `SparseTensor` of rank `2`, the strings split according to the delimiter.
251 return sparse_tensor.SparseTensor(indices, values, shape)
256 """Split elements of `source` based on `sep` into a `SparseTensor`.
259 element of `source` based on `sep` and return a `SparseTensor`
291 A `SparseTensor` of rank `2`, the strings split according to the delimiter.
305 return sparse_tensor.SparseTensor(indices, values, shape)
|
| confusion_matrix.py | 193 cm_sparse = sparse_tensor.SparseTensor(
|
| control_flow_grad.py | 166 if not isinstance(grad, (ops.IndexedSlices, sparse_tensor.SparseTensor)):
226 # TODO(yuanbyu, lukasr): Add support for SparseTensor.
|
| map_fn.py | 63 To apply a functional operation to the nonzero elements of a SparseTensor
68 result = SparseTensor(input.indices, fn(input.values), input.dense_shape)
74 result = SparseTensor(
125 `fn` and `dtype` do not match, or if elems is a SparseTensor.
151 if isinstance(elems, sparse_tensor.SparseTensor):
154 " SparseTensor(input.indices, fn(input.values), input.dense_shape) or "
155 " SparseTensor(input.indices, map_fn(fn, input.values), "
|
| math_ops.py | 234 x: A `Tensor` or `SparseTensor` of type `float16`, `float32`, `float64`,
239 A `Tensor` or `SparseTensor` the same size and type as `x` with absolute
351 x: A `Tensor` or `SparseTensor`. Must be one of the following types: `half`,
356 A `Tensor` or `SparseTensor`, respectively. Has the same type as `x`.
615 (in case of `SparseTensor` or `IndexedSlices`) to `dtype`.
634 x: A `Tensor` or `SparseTensor` or `IndexedSlices` of numeric type. It could
643 A `Tensor` or `SparseTensor` or `IndexedSlices` with same shape as `x` and
654 if isinstance(x, sparse_tensor.SparseTensor):
656 x = sparse_tensor.SparseTensor(x.indices, values_cast, x.dense_shape)
713 x: A `Tensor` or `SparseTensor` or `IndexedSlices`
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| input.py | 398 rank: The statically known rank of the `SparseTensor`.
487 restore the associated `SparseTensor` objects. In this case, the sparse
496 tensor_list: List of `Tensor` and `SparseTensor` objects.
513 if not isinstance(t, sparse_tensor.SparseTensor):
519 # the operation used to store the SparseTensor.
525 if not isinstance(t, sparse_tensor.SparseTensor):
612 has_st = any(isinstance(x, sparse_tensor.SparseTensor) for x in tensors)
615 x.values if isinstance(x, sparse_tensor.SparseTensor)
620 sparse_tensor.SparseTensor(indices=with_deps(x.indices),
623 if isinstance(x, sparse_tensor.SparseTensor)
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| embedding_ops_test.py | 666 sp_ids = sparse_tensor.SparseTensor(
670 sp_weights = sparse_tensor.SparseTensor(
776 sp_ids = sparse_tensor.SparseTensor(
780 sp_weights = sparse_tensor.SparseTensor(
[all...] |
| sparse_tensors_map_ops_test.py | 48 return sparse_tensor_lib.SparseTensor(
173 r"SparseTensor\[1\] was: 3 but rank of SparseTensor\[1\] is: 4"):
186 with self.assertRaisesOpError(r"Unable to find SparseTensor: 10"):
208 st = sparse_tensor_lib.SparseTensor(indices, values, shape)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| quantile_ops.py | 173 if isinstance(column, sparse_tensor.SparseTensor):
|
| /external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/ |
| masked_matmul_benchmark.py | 59 unordered_mask = sparse_tensor.SparseTensor(indices, values, mask_shape)
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| factorization_ops.py | 680 sp_input: A SparseTensor representing a subset of rows of the full input
681 in any order. Please note that this SparseTensor must retain the
713 sp_input: A SparseTensor representing a subset of columns of the full
752 sp_input: A SparseTensor representing a set of rows. Please note that the
753 column indices of this SparseTensor must match the model column feature
785 sp_input: A SparseTensor representing a set of columns. Please note that
786 the row indices of this SparseTensor must match the model row feature
[all...] |
| factorization_ops_test.py | 146 sp_input=sparse_tensor.SparseTensor(
681 inp = sparse_tensor.SparseTensor(indices, values, [rows, cols])
711 inp = sparse_tensor.SparseTensor(indices, values, [rows, cols])
749 inp = sparse_tensor.SparseTensor(indices, values, [rows, cols])
[all...] |
| /external/tensorflow/tensorflow/contrib/kernel_methods/python/ |
| kernel_estimators_test.py | 199 sparse_tensor.SparseTensor(
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| tensor_signature.py | 64 if isinstance(tensor, sparse_tensor.SparseTensor):
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| utils_test.py | 104 self.assertIsInstance(actual, sparse_tensor.SparseTensor)
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| feature_column.py | 263 this dict. Values can be a `SparseTensor` or a `Tensor` depends on
383 `indicator_column`s. To be specific, assume the input as `SparseTensor` looks
414 this dict. Values are `Tensor` or `SparseTensor` depending on
731 this dict. Values can be a `SparseTensor` or a `Tensor` depends on
736 A `dict` mapping `_FeatureColumn` to `Tensor` and `SparseTensor` values.
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| feature_column.py | 316 ("column_name", SparseTensor) pair.
444 if not isinstance(input_tensor, sparse_tensor_py.SparseTensor):
506 return sparse_tensor_py.SparseTensor(input_tensor.indices, sparse_id_values,
597 sparse_tensor_py.SparseTensor(input_tensor.indices,
604 return sparse_tensor_py.SparseTensor(
705 sparse_string_tensor = sparse_tensor_py.SparseTensor(st.indices,
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| graph_io.py | 523 - A dict of `Tensor` or `SparseTensor` objects for each in `features`.
606 - A dict of `Tensor` or `SparseTensor` objects for each in `features`.
654 parsed_features: A dict of string key to `Tensor` or `SparseTensor` objects.
668 - A dict of string key to `Tensor` or `SparseTensor` objects corresponding
691 if isinstance(tensor, sparse_tensor.SparseTensor):
742 dequeued_parsed_features[key] = sparse_tensor.SparseTensor(
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| subscribe_test.py | 289 sparse_tensor_1 = sparse_tensor.SparseTensor(
291 sparse_tensor_2 = sparse_tensor.SparseTensor(
|
