| /external/tensorflow/tensorflow/python/kernel_tests/ |
| sparse_to_dense_op_py_test.py | 29 def _SparseToDense(sparse_indices,
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/tensor_forest/kernels/v4/ |
| input_data.cc | 105 const Tensor& sparse_indices,
111 if (sparse_indices.shape().dims() == 2) {
113 new SparseIndicesStorageType(sparse_indices.tensor<int64, 2>()));
|
| input_data.h | 69 void set_input_tensors(const Tensor& dense, const Tensor& sparse_indices,
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| 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/core/util/ |
| example_proto_fast_parsing.h | 67 std::vector<Tensor> sparse_indices; member in struct:tensorflow::example::Result
|
| example_proto_fast_parsing.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| sparse_column_iterable.h | 115 const TTypes<int64>::ConstMatrix& sparse_indices() const { return ix_; } function in class:tensorflow::boosted_trees::utils::SparseColumnIterable
|
| examples_iterable.h | 117 .sparse_indices();
|
| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_to_dense_op_test.cc | 55 // sparse_indices
74 // sparse_indices
93 // sparse_indices
112 // sparse_indices
134 // sparse_indices
156 // sparse_indices
178 // sparse_indices
218 Tensor sparse_indices(DT_INT32, TensorShape({N, NDIM}));
226 auto sparse_indices_t = sparse_indices.matrix<int32>();
233 {&sparse_indices, &output_shape, &sparse_values, &default_value})
[all...] |
| sdca_ops_test.cc | 160 std::vector<NodeBuilder::NodeOut> sparse_indices; local
163 sparse_indices.push_back(
220 .Input(sparse_indices)
|
| example_parsing_ops.cc | 147 OpOutputList sparse_indices; variable
151 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_indices", &sparse_indices));
158 sparse_indices.set(d, result.sparse_indices[d]);
243 OpOutputList sparse_indices; variable
247 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_indices", &sparse_indices));
254 sparse_indices.set(d, result.sparse_indices[d])
[all...] |
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| 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/compiler/xla/ |
| literal_util.h | 109 const SparseIndexArray* sparse_indices(
111 SparseIndexArray* sparse_indices(const ShapeIndex& shape_index = {});
689 SparseIndexArray* sparse_indices() const { return sparse_indices_; } function in class:xla::Literal::Piece
690 void set_sparse_indices(SparseIndexArray* sparse_indices) {
691 sparse_indices_ = sparse_indices;
[all...] |
| literal_util.cc | 121 delete piece.sparse_indices();
174 const SparseIndexArray* Literal::sparse_indices( function in class:xla::Literal
176 return piece(shape_index).sparse_indices();
179 SparseIndexArray* Literal::sparse_indices(const ShapeIndex& shape_index) { function in class:xla::Literal
180 return piece(shape_index).sparse_indices();
271 dest_piece.set_sparse_indices(src_piece.sparse_indices());
429 delete dest_piece.sparse_indices();
430 dest_piece.set_sparse_indices(src_piece.sparse_indices());
[all...] |
| xla_data.proto | 344 repeated int64 sparse_indices = 14;
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ |
| 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/contrib/tensor_forest/kernels/ |
| tree_utils.cc | 555 GetFeatureFnType GetSparseFunctor(const Tensor& sparse_indices,
557 if (sparse_indices.shape().dims() == 2) {
558 const auto indices = sparse_indices.matrix<int64>();
|
| tree_utils.h | 161 GetFeatureFnType GetSparseFunctor(const Tensor& sparse_indices,
|
| /frameworks/base/libs/androidfw/ |
| LoadedArsc.cpp | 225 const ResTable_sparseTypeEntry* sparse_indices = local
228 const ResTable_sparseTypeEntry* sparse_indices_end = sparse_indices + entry_count;
230 std::lower_bound(sparse_indices, sparse_indices_end, entry_index,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| quantile_ops.cc | 53 const char* const kSparseIndicesName = "sparse_indices";
416 const auto sparse_indices =
428 const int64 example_id = sparse_indices(j, 0);
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| parsing_ops.py | 715 (sparse_indices, sparse_values, sparse_shapes, dense_values) = outputs
719 in zip(sparse_indices, sparse_values, sparse_shapes)]
[all...] |
| nn_impl.py | [all...] |
| sparse_ops.py | 739 def sparse_to_dense(sparse_indices,
750 # If sparse_indices is scalar
751 dense[i] = (i == sparse_indices ? sparse_values : default_value)
753 # If sparse_indices is a vector, then for each i
754 dense[sparse_indices[i]] = sparse_values[i]
756 # If sparse_indices is an n by d matrix, then for each i in [0, n)
757 dense[sparse_indices[i][0], ..., sparse_indices[i][d-1]] = sparse_values[i]
768 sparse_indices: A 0-D, 1-D, or 2-D `Tensor` of type `int32` or `int64`.
769 `sparse_indices[i]` contains the complete index where `sparse_values[i]
[all...] |
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| factorization_ops_test.py | 120 sparse_indices = [[0, 1], [2, 3], [4, 1]]
125 # sparse_indices has three unique rows and two unique columns
147 indices=sparse_indices,
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
