/external/tensorflow/tensorflow/core/kernels/data/ |
D | cache_dataset_ops.cc | 116 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 …]
|
D | sparse_tensor_slice_dataset_op.cc | 94 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()
|
D | zip_dataset_op.cc | 108 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()
|
D | unique_dataset_op.cc | 94 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()
|
D | repeat_dataset_op.cc | 92 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()
|
D | tensor_slice_dataset_op.cc | 112 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()
|
D | stats_dataset_ops.cc | 93 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()
|
D | filter_dataset_op.cc | 123 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()
|
D | take_dataset_op.cc | 92 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()
|
D | skip_dataset_op.cc | 91 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()
|
D | window_dataset.cc | 50 std::vector<Tensor>* out_tensors, in GetNextInternal() argument 57 *out_tensors = dataset()->elements_[i_++]; in GetNextInternal()
|
D | tensor_dataset_op.cc | 93 std::vector<Tensor>* out_tensors, in GetNextInternal() argument 97 *out_tensors = dataset()->tensors_; in GetNextInternal()
|
D | reader_dataset_ops.cc | 133 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()
|
D | range_dataset_op.cc | 95 std::vector<Tensor>* out_tensors, in GetNextInternal() argument 105 out_tensors->emplace_back(std::move(value_tensor)); in GetNextInternal()
|
D | sql_dataset_ops.cc | 120 std::vector<Tensor>* out_tensors, in GetNextInternal() argument 135 return query_connection_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
|
D | random_dataset_op.cc | 99 std::vector<Tensor>* out_tensors, in GetNextInternal() argument 104 out_tensors->emplace_back(std::move(value_tensor)); in GetNextInternal()
|
D | concatenate_dataset_op.cc | 103 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/ |
D | while_loop_test.cc | 59 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/ |
D | while_gradients_test.cc | 73 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/ |
D | ignore_errors_dataset_op.cc | 79 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/ |
D | main.cc | 117 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/ |
D | main.cc | 75 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/ |
D | python_api.md | 56 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/ |
D | sqlite_query_connection.cc | 53 std::vector<Tensor>* out_tensors, in GetNext() argument 63 out_tensors->emplace_back(std::move(tensor)); in GetNext()
|
/external/tensorflow/tensorflow/core/framework/ |
D | dataset.h | 333 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,
|