Searched defs:sparse_indices (Results 1 - 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/util/ |
H A D | example_proto_fast_parsing.h | 67 std::vector<Tensor> sparse_indices; member in struct:tensorflow::example::Result
|
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
H A D | input_data.cc | 105 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 D | sparse_column_iterable.h | 115 const TTypes<int64>::ConstMatrix& sparse_indices() const { return ix_; } function in class:tensorflow::boosted_trees::utils::SparseColumnIterable
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | sdca_ops_test.cc | 160 std::vector<NodeBuilder::NodeOut> sparse_indices; local 163 sparse_indices.push_back( 220 .Input(sparse_indices)
|
H A D | 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[ [all...] |
/external/tensorflow/tensorflow/compiler/xla/ |
H A D | 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()); 1004 CHECK_LT(sparse_element_number, p.sparse_indices()->index_count()); 1005 return p.sparse_indices() [all...] |
H A 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_; } 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 D | tree_utils.cc | 555 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