Home
last modified time | relevance | path

Searched refs:out_tensors (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Dcache_dataset_ops.cc116 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
133 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
134 if (*end_of_sequence && out_tensors->empty()) { in GetNextInternal()
139 if (out_tensors->size() != dataset()->num_tensors_) { in GetNextInternal()
142 dataset()->num_tensors_, " got: ", out_tensors->size()); in GetNextInternal()
145 for (const Tensor& t : *out_tensors) { in GetNextInternal()
218 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
228 out_tensors->clear(); in GetNextInternal()
229 out_tensors->resize(dataset()->num_tensors_); in GetNextInternal()
234 out_tensors->clear(); in GetNextInternal()
[all …]
Dsparse_tensor_slice_dataset_op.cc94 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
102 out_tensors->clear(); in GetNextInternal()
103 out_tensors->reserve(3); in GetNextInternal()
135 out_tensors->push_back(std::move(next_indices_)); in GetNextInternal()
136 out_tensors->push_back(std::move(next_values_)); in GetNextInternal()
137 out_tensors->push_back(dense_shape_); in GetNextInternal()
143 out_tensors->push_back(Tensor(DT_INT64, TensorShape({0, rank - 1}))); in GetNextInternal()
144 out_tensors->push_back(Tensor(DataTypeToEnum<T>::value, {0})); in GetNextInternal()
145 out_tensors->push_back(dense_shape_); in GetNextInternal()
Dzip_dataset_op.cc108 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
115 out_tensors->clear(); in GetNextInternal()
116 out_tensors->reserve(dataset()->output_dtypes().size()); in GetNextInternal()
124 out_tensors->insert(out_tensors->end(), input_tensors.begin(), in GetNextInternal()
128 out_tensors->clear(); in GetNextInternal()
Dunique_dataset_op.cc94 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
100 out_tensors->clear(); in GetNextInternal()
102 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
106 DCHECK_EQ(1, out_tensors->size()); in GetNextInternal()
107 saw_new_value = unique_elements_.insert((*out_tensors)[0]).second; in GetNextInternal()
Drepeat_dataset_op.cc92 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
116 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
125 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
174 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
184 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
Dtensor_slice_dataset_op.cc112 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
116 out_tensors->clear(); in GetNextInternal()
117 out_tensors->reserve(dataset()->tensors_.size()); in GetNextInternal()
123 out_tensors->emplace_back(std::move(t_slice)); in GetNextInternal()
Dstats_dataset_ops.cc93 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
97 Status s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
189 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
192 Status s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
196 for (const Tensor& t : *out_tensors) { in GetNextInternal()
Dfilter_dataset_op.cc123 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
138 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
150 ctx, *out_tensors, &result)); in GetNextInternal()
160 out_tensors->clear(); in GetNextInternal()
Dtake_dataset_op.cc92 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
117 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
126 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
Dskip_dataset_op.cc91 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
116 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
144 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
Dwindow_dataset.cc50 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
57 *out_tensors = dataset()->elements_[i_++]; in GetNextInternal()
Dtensor_dataset_op.cc93 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
97 *out_tensors = dataset()->tensors_; in GetNextInternal()
Dreader_dataset_ops.cc133 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
146 out_tensors->emplace_back(std::move(line_tensor)); in GetNextInternal()
374 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
389 out_tensors->emplace_back(std::move(record_tensor)); in GetNextInternal()
586 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
595 out_tensors->emplace_back(std::move(result_tensor)); in GetNextInternal()
Drange_dataset_op.cc95 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
105 out_tensors->emplace_back(std::move(value_tensor)); in GetNextInternal()
Dsql_dataset_ops.cc120 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
135 return query_connection_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
Drandom_dataset_op.cc99 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
104 out_tensors->emplace_back(std::move(value_tensor)); in GetNextInternal()
Dconcatenate_dataset_op.cc103 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
112 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
/external/tensorflow/tensorflow/cc/ops/
Dwhile_loop_test.cc59 std::vector<Tensor> out_tensors; in Run() local
60 TF_ASSERT_OK(session.Run(feeds, outputs_, &out_tensors)); in Run()
61 ASSERT_EQ(out_tensors.size(), outputs_.size()); in Run()
63 DCHECK_EQ(expected_output_values.size(), out_tensors.size()); in Run()
64 for (int i = 0; i < out_tensors.size(); ++i) { in Run()
66 out_tensors[i], test::AsTensor<T>({expected_output_values[i]}, {})); in Run()
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients_test.cc73 std::vector<Tensor> out_tensors; in Run() local
75 &out_tensors, run_metadata)); in Run()
76 ASSERT_EQ(out_tensors.size(), grad_outputs_.size()); in Run()
78 DCHECK_EQ(expected_grad_values.size(), out_tensors.size()); in Run()
79 for (int i = 0; i < out_tensors.size(); ++i) { in Run()
81 out_tensors[i], test::AsTensor<T>({expected_grad_values[i]}, {})); in Run()
/external/tensorflow/tensorflow/contrib/data/kernels/
Dignore_errors_dataset_op.cc79 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
87 Status s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
89 out_tensors->clear(); in GetNextInternal()
90 s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
/external/tensorflow/tensorflow/examples/label_image/
Dmain.cc117 std::vector<Tensor>* out_tensors) { in ReadTensorFromImageFile() argument
179 TF_RETURN_IF_ERROR(session->Run({inputs}, {output_name}, {}, out_tensors)); in ReadTensorFromImageFile()
221 std::vector<Tensor> out_tensors; in GetTopLabels() local
223 {}, &out_tensors)); in GetTopLabels()
224 *scores = out_tensors[0]; in GetTopLabels()
225 *indices = out_tensors[1]; in GetTopLabels()
/external/tensorflow/tensorflow/examples/multibox_detector/
Dmain.cc75 std::vector<Tensor>* out_tensors) { in ReadTensorFromImageFile() argument
128 session->Run({}, {output_name, original_name}, {}, out_tensors)); in ReadTensorFromImageFile()
198 std::vector<Tensor> out_tensors; in GetTopDetections() local
200 {}, &out_tensors)); in GetTopDetections()
201 *scores = out_tensors[0]; in GetTopDetections()
202 *indices = out_tensors[1]; in GetTopDetections()
/external/tensorflow/tensorflow/contrib/lite/toco/g3doc/
Dpython_api.md56 out_tensors = [out]
58 sess, sess.graph_def, map(canonical_name, out_tensors))
60 frozen_graphdef, [img], out_tensors)
/external/tensorflow/tensorflow/core/kernels/data/sql/
Dsqlite_query_connection.cc53 std::vector<Tensor>* out_tensors, in GetNext() argument
63 out_tensors->emplace_back(std::move(tensor)); in GetNext()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h333 virtual Status GetNext(IteratorContext* ctx, std::vector<Tensor>* out_tensors,
514 Status GetNext(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in GetNext() argument
517 Status s = GetNextInternal(ctx, out_tensors, end_of_sequence); in GetNext()
537 std::vector<Tensor>* out_tensors,

12