| /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_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"
|
| D | api_def_TakeManySparseFromTensorsMap.pbtxt | 11 name: "sparse_indices"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_to_dense_op_py_test.py | 29 def _SparseToDense(sparse_indices, argument
35 sparse_indices,
125 sparse_indices=[[1], [10]],
135 sparse_indices=[[1], [10]],
145 sparse_indices=[[1], [1]],
153 sparse_indices=[[1], [1]],
163 sparse_indices=[[2], [1]],
171 sparse_indices=[[2], [1]],
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| D | sdca_optimizer.py | 111 sparse_indices = array_ops.where(
114 sparse_values = array_ops.gather_nd(dense_tensor, sparse_indices)
120 value=sparse_indices, num_or_size_splits=2, axis=1)[0], [-1]),
123 value=sparse_indices, num_or_size_splits=2, axis=1)[1], [-1]),
|
| /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 | 69 void set_input_tensors(const Tensor& dense, const Tensor& sparse_indices,
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sdca_ops.py | 320 sparse_indices = []
324 sparse_indices.append(
328 sparse_weights.append(array_ops.gather(w, sparse_indices[-1]))
338 sparse_indices,
354 sparse_indices, sfw):
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ |
| D | tensor_forest.py | 601 sparse_indices = []
605 sparse_indices = sparse_features.indices
615 sparse_indices,
632 sparse_indices,
661 sparse_indices = []
665 sparse_indices = sparse_features.indices
674 sparse_indices,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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()
|
| 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/compiler/xla/ |
| D | literal_util.h | 109 const SparseIndexArray* sparse_indices(
111 SparseIndexArray* sparse_indices(const ShapeIndex& shape_index = {});
689 SparseIndexArray* sparse_indices() const { return sparse_indices_; } in sparse_indices() function
690 void set_sparse_indices(SparseIndexArray* sparse_indices) { in set_sparse_indices() argument
691 sparse_indices_ = sparse_indices; in set_sparse_indices()
1131 int64 last_element = p.sparse_indices()->index_count(); in AppendSparseElement()
1133 p.sparse_indices()->Append(multi_index); in AppendSparseElement()
1242 *this->root_piece().sparse_indices() = std::move(indices); in PopulateSparse()
1246 this->root_piece().sparse_indices()->SortWithValues(root_data); in PopulateSparse()
1248 DCHECK(this->root_piece().sparse_indices()->Validate(shape())); in PopulateSparse()
|
| D | literal_util.cc | 121 delete piece.sparse_indices(); in DeallocateBuffers()
174 const SparseIndexArray* Literal::sparse_indices( in sparse_indices() function in xla::Literal
176 return piece(shape_index).sparse_indices(); in sparse_indices()
179 SparseIndexArray* Literal::sparse_indices(const ShapeIndex& shape_index) { in sparse_indices() function in xla::Literal
180 return piece(shape_index).sparse_indices(); in sparse_indices()
271 dest_piece.set_sparse_indices(src_piece.sparse_indices()); in DecomposeTuple()
429 delete dest_piece.sparse_indices(); in MoveFrom()
430 dest_piece.set_sparse_indices(src_piece.sparse_indices()); in MoveFrom()
1004 CHECK_LT(sparse_element_number, p.sparse_indices()->index_count()); in GetSparseIndex()
1005 return p.sparse_indices()->At(sparse_element_number); in GetSparseIndex()
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.h | 67 std::vector<Tensor> sparse_indices; member
|
| D | example_proto_fast_parsing.cc | 1003 result->sparse_indices.emplace_back(DT_INT64, indices_shape); in FastParseExample()
1004 Tensor* indices = &result->sparse_indices.back(); in FastParseExample()
1187 result->sparse_indices.emplace_back(); in FastParseSingleExample()
1377 Tensor* out_indices = &result->sparse_indices[d]; in FastParseSingleExample()
1448 result->sparse_indices[d] = Tensor(DT_INT64, TensorShape({0, 1})); in FastParseSingleExample()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_ops.py | 715 (sparse_indices, sparse_values, sparse_shapes, dense_values) = outputs
719 in zip(sparse_indices, sparse_values, sparse_shapes)]
1406 (sparse_indices, sparse_values, sparse_shapes, dense_values) = outputs
1411 shape) in zip(sparse_indices, sparse_values, sparse_shapes)
|
| /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()
|
| D | tree_utils.h | 161 GetFeatureFnType GetSparseFunctor(const Tensor& 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
|
| D | examples_iterable.h | 117 .sparse_indices();
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 416 const auto sparse_indices = in Compute() local
428 const int64 example_id = sparse_indices(j, 0); in Compute()
800 const auto sparse_indices = in Compute() local
808 const int64 example_id = sparse_indices(j, 0); in Compute()
|