/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(
|