Searched defs:sparse_indices (Results 1 - 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/util/
H A Dexample_proto_fast_parsing.h67 std::vector<Tensor> sparse_indices; member in struct:tensorflow::example::Result
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
H A Dinput_data.cc105 const Tensor& sparse_indices,
111 if (sparse_indices.shape().dims() == 2) {
113 new SparseIndicesStorageType(sparse_indices.tensor<int64, 2>()));
104 set_input_tensors(const Tensor& dense, const Tensor& sparse_indices, const Tensor& sparse_values, const Tensor& sparse_shape) argument
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
H A Dsparse_column_iterable.h115 const TTypes<int64>::ConstMatrix& sparse_indices() const { return ix_; } function in class:tensorflow::boosted_trees::utils::SparseColumnIterable
/external/tensorflow/tensorflow/core/kernels/
H A Dsdca_ops_test.cc160 std::vector<NodeBuilder::NodeOut> sparse_indices; local
163 sparse_indices.push_back(
220 .Input(sparse_indices)
H A Dexample_parsing_ops.cc147 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[
[all...]
/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral_util.cc121 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());
1004 CHECK_LT(sparse_element_number, p.sparse_indices()->index_count());
1005 return p.sparse_indices()
[all...]
H A Dliteral_util.h109 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) { argument
691 sparse_indices_ = sparse_indices;
1131 int64 last_element = p.sparse_indices()->index_count();
1133 p.sparse_indices()->Append(multi_index);
1242 *this->root_piece().sparse_indices() = std::move(indices);
1246 this->root_piece().sparse_indices()->SortWithValues(root_data);
1248 DCHECK(this->root_piece().sparse_indices()
[all...]
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
H A Dtree_utils.cc555 GetFeatureFnType GetSparseFunctor(const Tensor& sparse_indices, argument
557 if (sparse_indices.shape().dims() == 2) {
558 const auto indices = sparse_indices.matrix<int64>();

Completed in 220 milliseconds