| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseToDense.pbtxt | 4 name: "sparse_indices"
6 0-D, 1-D, or 2-D. `sparse_indices[i]` contains the complete
19 1-D. Values corresponding to each row of `sparse_indices`,
27 `sparse_indices`.
48 # If sparse_indices is scalar
49 dense[i] = (i == sparse_indices ? sparse_values : default_value)
51 # If sparse_indices is a vector, then for each i
52 dense[sparse_indices[i]] = sparse_values[i]
54 # If sparse_indices is an n by d matrix, then for each i in [0, n)
55 dense[sparse_indices[i][0], ..., sparse_indices[i][d-1]] = sparse_values[i]
|
| D | api_def_AddManySparseToTensorsMap.pbtxt | 4 name: "sparse_indices"
7 `sparse_indices[:, 0]` must be ordered values in `[0, N)`.
45 A `SparseTensor` of rank `R` is represented by three tensors: `sparse_indices`,
48 ```sparse_indices.shape[1] == sparse_shape.shape[0] == R```
51 having a first `sparse_indices` column taking values between `[0, N)`, where
|
| D | api_def_RaggedTensorToSparse.pbtxt | 13 name: "sparse_indices"
39 output=SparseTensor(indices=sparse_indices, values=sparse_values,
|
| D | api_def_AddSparseToTensorsMap.pbtxt | 4 name: "sparse_indices"
43 A `SparseTensor` is represented by three tensors: `sparse_indices`,
|
| D | api_def_SerializeSparse.pbtxt | 4 name: "sparse_indices"
|
| D | api_def_SerializeManySparse.pbtxt | 4 name: "sparse_indices"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_to_dense_op_py_test.py | 30 def _SparseToDense(sparse_indices, argument
36 sparse_indices,
139 sparse_indices=[[1], [10]],
149 sparse_indices=[[1], [10]],
160 sparse_indices=[[1], [1]],
168 sparse_indices=[[1], [1]],
179 sparse_indices=[[2], [1]],
187 sparse_indices=[[2], [1]],
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_sparse_kernel.cc | 69 auto sparse_indices = sparse_indices_out->tensor<int64, 2>(); in Compute() local
105 sparse_indices(next_index, dim++) = index; in Compute()
107 sparse_indices(next_index, dim++) = i; // index_middle in Compute()
109 sparse_indices(next_index, dim++) = index; in Compute()
|
| D | example_parsing_ops.cc | 147 OpOutputList sparse_indices; in Compute() local
151 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_indices", &sparse_indices)); in Compute()
158 sparse_indices.set(d, result.sparse_indices[d]); in Compute()
243 OpOutputList sparse_indices; in Compute() local
247 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_indices", &sparse_indices)); in Compute()
254 sparse_indices.set(d, result.sparse_indices[d]); in Compute()
408 context_sparse_indices.set(d, context_result.sparse_indices[d]); in Compute()
413 feature_list_sparse_indices.set(d, feature_list_result.sparse_indices[d]); in Compute()
|
| D | sparse_to_dense_op_test.cc | 218 Tensor sparse_indices(DT_INT32, TensorShape({N, NDIM})); in BM_SparseToDense() local
226 auto sparse_indices_t = sparse_indices.matrix<int32>(); in BM_SparseToDense()
233 {&sparse_indices, &output_shape, &sparse_values, &default_value}) { in BM_SparseToDense()
|
| D | sdca_ops_test.cc | 160 std::vector<NodeBuilder::NodeOut> sparse_indices; in GetGraphs() local
163 sparse_indices.push_back( in GetGraphs()
220 .Input(sparse_indices) in GetGraphs()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | input_data.cc | 105 const Tensor& sparse_indices, in set_input_tensors() argument
111 if (sparse_indices.shape().dims() == 2) { in set_input_tensors()
113 new SparseIndicesStorageType(sparse_indices.tensor<int64, 2>())); in set_input_tensors()
|
| D | input_data.h | 73 void set_input_tensors(const Tensor& dense, const Tensor& sparse_indices,
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| D | sdca_optimizer.py | 125 sparse_indices = array_ops.where(
128 sparse_values = array_ops.gather_nd(dense_tensor, sparse_indices)
134 value=sparse_indices, num_or_size_splits=2, axis=1)[0], [-1]),
137 value=sparse_indices, num_or_size_splits=2, axis=1)[1], [-1]),
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.h | 82 const SparseIndexArray* sparse_indices(
353 SparseIndexArray* sparse_indices() const { return sparse_indices_; } in sparse_indices() function
354 void set_sparse_indices(SparseIndexArray* sparse_indices) { in set_sparse_indices() argument
355 sparse_indices_ = sparse_indices; in set_sparse_indices()
371 ? sparse_indices()->index_count() in element_count()
565 SparseIndexArray* sparse_indices(const ShapeIndex& shape_index = {});
970 int64 last_element = p.sparse_indices()->index_count(); in AppendSparseElement()
972 p.sparse_indices()->Append(multi_index); in AppendSparseElement()
1072 *this->root_piece().sparse_indices() = std::move(indices); in PopulateSparse()
1075 this->root_piece().sparse_indices()->SortWithValues(root_data); in PopulateSparse()
[all …]
|
| D | literal.cc | 168 delete piece->sparse_indices(); in DeallocateBuffers()
198 const SparseIndexArray* LiteralBase::sparse_indices( in sparse_indices() function in xla::LiteralBase
200 return piece(shape_index).sparse_indices(); in sparse_indices()
203 SparseIndexArray* MutableLiteralBase::sparse_indices( in sparse_indices() function in xla::MutableLiteralBase
205 return piece(shape_index).sparse_indices(); in sparse_indices()
357 dest_piece->set_sparse_indices(src_piece.sparse_indices()); in DecomposeTuple()
492 delete dest_piece.sparse_indices(); in MoveFrom()
493 dest_piece.set_sparse_indices(src_piece.sparse_indices()); in MoveFrom()
969 CHECK_LT(sparse_element_number, p.sparse_indices()->index_count()); in GetSparseIndex()
970 return p.sparse_indices()->At(sparse_element_number); in GetSparseIndex()
[all …]
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ |
| D | tensor_forest.py | 613 sparse_indices = []
617 sparse_indices = sparse_features.indices
627 sparse_indices,
644 sparse_indices,
673 sparse_indices = []
677 sparse_indices = sparse_features.indices
686 sparse_indices,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sparse_to_dense_op_test.py | 30 def _SparseToDense(sparse_indices, argument
36 feed_dict = {feed_sparse_indices: sparse_indices}
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 317 sparse_indices, output_shape, _, _ = op.inputs
319 sparse_values_grad = array_ops.gather_nd(grad, sparse_indices)
323 array_ops.zeros_like(sparse_indices),
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sdca_ops.py | 417 sparse_indices = []
432 sparse_indices.append(sparse_idx)
512 sparse_indices,
531 sparse_indices,
549 sparse_indices, sfw)):
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.h | 87 std::vector<Tensor> sparse_indices; member
|
| D | example_proto_fast_parsing.cc | 1098 result->sparse_indices.emplace_back(DT_INT64, indices_shape); in FastParseExample()
1099 Tensor* indices = &result->sparse_indices.back(); in FastParseExample()
1288 result->sparse_indices.emplace_back(); in FastParseSingleExample()
1494 Tensor* out_indices = &result->sparse_indices[d]; in FastParseSingleExample()
1565 result->sparse_indices[d] = Tensor(DT_INT64, TensorShape({0, 1})); in FastParseSingleExample()
1992 context_result->sparse_indices.resize(context_config.sparse.size()); in FastParseSequenceExample()
1996 feature_list_result->sparse_indices.resize(feature_list_config.sparse.size()); in FastParseSequenceExample()
2125 context_result->sparse_indices[t] = Tensor(DT_INT64, indices_shape); in FastParseSequenceExample()
2147 int64* out_indices = context_result->sparse_indices[t].flat<int64>().data(); in FastParseSequenceExample()
2323 feature_list_result->sparse_indices[t] = Tensor(DT_INT64, indices_shape); in FastParseSequenceExample()
[all …]
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | tree_utils.cc | 555 GetFeatureFnType GetSparseFunctor(const Tensor& sparse_indices, in GetSparseFunctor() argument
557 if (sparse_indices.shape().dims() == 2) { in GetSparseFunctor()
558 const auto indices = sparse_indices.matrix<int64>(); in GetSparseFunctor()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 707 sparse_indices = []
711 sparse_indices.append(i)
715 return dense_values, dense_indices, sparse_values, sparse_indices
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | sparse_column_iterable.h | 115 const TTypes<int64>::ConstMatrix& sparse_indices() const { return ix_; } in sparse_indices() function
|