| /external/tensorflow/tensorflow/contrib/ignite/kernels/dataset/ |
| D | ignite_binary_object_parser.cc | 25 std::vector<Tensor>* out_tensors, in Parse() argument
35 out_tensors->emplace_back(cpu_allocator(), DT_UINT8, TensorShape({})); in Parse()
36 out_tensors->back().scalar<uint8>()() = ParseByte(ptr); in Parse()
40 out_tensors->emplace_back(cpu_allocator(), DT_INT16, TensorShape({})); in Parse()
41 out_tensors->back().scalar<int16>()() = ParseShort(ptr); in Parse()
45 out_tensors->emplace_back(cpu_allocator(), DT_UINT16, TensorShape({})); in Parse()
46 out_tensors->back().scalar<uint16>()() = ParseUnsignedShort(ptr); in Parse()
50 out_tensors->emplace_back(cpu_allocator(), DT_INT32, TensorShape({})); in Parse()
51 out_tensors->back().scalar<int32>()() = ParseInt(ptr); in Parse()
55 out_tensors->emplace_back(cpu_allocator(), DT_INT64, TensorShape({})); in Parse()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op_test.cc | 133 std::vector<Tensor> out_tensors; in TEST() local
134 TF_ASSERT_OK(session.Run(feeds, {Output(while_node)}, &out_tensors)); in TEST()
135 ASSERT_EQ(out_tensors.size(), 1); in TEST()
136 EXPECT_EQ(out_tensors[0].scalar<int>()(), 16); in TEST()
141 std::vector<Tensor> out_tensors; in TEST() local
142 TF_ASSERT_OK(session.Run(feeds, {Output(while_node)}, &out_tensors)); in TEST()
143 ASSERT_EQ(out_tensors.size(), 1); in TEST()
144 EXPECT_EQ(out_tensors[0].scalar<int>()(), 12); in TEST()
225 std::vector<Tensor> out_tensors; in TEST() local
227 feeds, {Output(while_node, 0), Output(while_node, 1)}, &out_tensors)); in TEST()
[all …]
|
| D | lower_if_op_test.cc | 125 std::vector<Tensor> out_tensors; in TEST() local
126 TF_ASSERT_OK(session.Run(feeds, {Output(written_if)}, &out_tensors)); in TEST()
127 EXPECT_EQ(out_tensors.size(), 1); in TEST()
128 EXPECT_EQ(out_tensors[0].scalar<int>()(), 40); in TEST()
134 std::vector<Tensor> out_tensors; in TEST() local
135 TF_ASSERT_OK(session.Run(feeds, {Output(written_if)}, &out_tensors)); in TEST()
136 EXPECT_EQ(out_tensors.size(), 1); in TEST()
137 EXPECT_EQ(out_tensors[0].scalar<int>()(), 20); in TEST()
233 std::vector<Tensor> out_tensors; in TEST() local
234 TF_ASSERT_OK(session.Run(feeds, {Output(read)}, &out_tensors)); in TEST()
[all …]
|
| D | lower_if_while_test.cc | 174 std::vector<Tensor> out_tensors; in TEST() local
175 TF_ASSERT_OK(session.Run(feeds, {Output(while_node, 2)}, &out_tensors)); in TEST()
176 ASSERT_EQ(out_tensors.size(), 1); in TEST()
177 EXPECT_EQ(out_tensors[0].scalar<int>()(), 8); in TEST()
184 std::vector<Tensor> out_tensors; in TEST() local
185 TF_ASSERT_OK(session.Run(feeds, {Output(while_node, 2)}, &out_tensors)); in TEST()
186 ASSERT_EQ(out_tensors.size(), 1); in TEST()
187 EXPECT_EQ(out_tensors[0].scalar<int>()(), 64); // a in TEST()
300 std::vector<Tensor> out_tensors; in TEST() local
301 TF_ASSERT_OK(session.Run(feeds, {Output(if_node)}, &out_tensors)); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model_test.py | 171 graph_def, in_tensors, out_tensors = convert_saved_model.freeze_saved_model(
178 return graph_def, in_tensors, out_tensors
183 _, in_tensors, out_tensors = self._convertSavedModel(saved_model_dir)
185 self.assertEqual(self._getArrayNames(out_tensors), ["add:0"])
192 _, in_tensors, out_tensors = self._convertSavedModel(saved_model_dir)
194 self.assertEqual(self._getArrayNames(out_tensors), ["add:0"])
235 _, in_tensors, out_tensors = self._convertSavedModel(
240 self.assertEqual(self._getArrayNames(out_tensors), ["add:0"])
247 _, in_tensors, out_tensors = self._convertSavedModel(
252 self.assertEqual(self._getArrayNames(out_tensors), ["add:0"])
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | csv_dataset_op.cc | 224 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
231 Status s = ReadRecord(ctx, out_tensors, select_all, in GetNextInternal()
235 if (out_tensors->size() != dataset()->out_type_.size()) { in GetNextInternal()
238 out_tensors->size(), " in record"); in GetNextInternal()
334 Status ReadRecord(IteratorContext* ctx, std::vector<Tensor>* out_tensors, in ReadRecord() argument
360 ParseOneField(ctx, out_tensors, &end_of_record, include)); in ReadRecord()
373 std::vector<Tensor>* out_tensors, in ParseOneField() argument
385 return FieldToOutput(ctx, StringPiece(), out_tensors); in ParseOneField()
397 return ParseQuotedField(ctx, out_tensors, end_of_record, include); in ParseOneField()
400 return ParseUnquotedField(ctx, out_tensors, end_of_record, include); in ParseOneField()
[all …]
|
| D | indexed_dataset_op.cc | 40 std::vector<Tensor>* out_tensors) const = 0;
113 std::vector<Tensor>* out_tensors) { in Get() argument
116 return captured->Get(std::move(ctx), index, out_tensors); in Get()
366 std::vector<Tensor> out_tensors; in Compute() local
368 materialized_resource->Get(IteratorContext(ctx), index, &out_tensors); in Compute()
379 for (size_t i = 0; i < out_tensors.size(); ++i) { in Compute()
381 ctx, expected_shapes[i].IsCompatibleWith(out_tensors[i].shape()), in Compute()
385 expected_shapes[i], ", got: ", out_tensors[i].shape(), ")")); in Compute()
386 OP_REQUIRES(ctx, out_tensors[i].dtype() == expected_types[i], in Compute()
390 ", got: ", out_tensors[i].dtype(), ")")); in Compute()
[all …]
|
| D | ignore_errors_dataset_op.cc | 86 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
94 Status s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
96 out_tensors->clear(); in GetNextInternal()
97 s = input_impl_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
|
| D | sql_dataset_op.cc | 139 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
146 return query_connection_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
172 std::vector<Tensor> out_tensors; in RestoreInternal() local
175 TF_RETURN_IF_ERROR(query_connection_->GetNext(ctx, &out_tensors, in RestoreInternal()
177 out_tensors.clear(); in RestoreInternal()
|
| D | unique_dataset_op.cc | 98 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
104 out_tensors->clear(); in GetNextInternal()
106 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
110 DCHECK_EQ(1, out_tensors->size()); in GetNextInternal()
111 saw_new_value = unique_elements_.insert((*out_tensors)[0]).second; in GetNextInternal()
|
| D | unbatch_dataset_op.cc | 98 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
108 out_tensors->clear(); in GetNextInternal()
109 out_tensors->reserve(tensors_.size()); in GetNextInternal()
111 out_tensors->emplace_back(ctx->allocator({}), tensors_[i].dtype(), in GetNextInternal()
114 &tensors_[i], &out_tensors->back(), current_index_)); in GetNextInternal()
|
| D | lmdb_dataset_op.cc | 89 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
94 out_tensors->emplace_back(ctx->allocator({}), DT_STRING, in GetNextInternal()
96 Tensor& key_tensor = out_tensors->back(); in GetNextInternal()
100 out_tensors->emplace_back(ctx->allocator({}), DT_STRING, in GetNextInternal()
102 Tensor& value_tensor = out_tensors->back(); in GetNextInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | cache_dataset_ops.cc | 132 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
135 return iterator_->GetNext(ctx, out_tensors, end_of_sequence); in GetNextInternal()
212 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
229 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
230 if (*end_of_sequence && out_tensors->empty()) { in GetNextInternal()
235 if (out_tensors->size() != dataset()->num_tensors_) { in GetNextInternal()
239 dataset()->num_tensors_, " got: ", out_tensors->size()); in GetNextInternal()
242 for (const Tensor& t : *out_tensors) { in GetNextInternal()
448 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
459 out_tensors->clear(); in GetNextInternal()
[all …]
|
| D | sparse_tensor_slice_dataset_op.cc | 97 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
105 out_tensors->clear(); in GetNextInternal()
106 out_tensors->reserve(3); in GetNextInternal()
138 out_tensors->push_back(std::move(next_indices_)); in GetNextInternal()
139 out_tensors->push_back(std::move(next_values_)); in GetNextInternal()
140 out_tensors->push_back(dense_shape_); in GetNextInternal()
146 out_tensors->push_back(Tensor(DT_INT64, TensorShape({0, rank - 1}))); in GetNextInternal()
147 out_tensors->push_back(Tensor(DataTypeToEnum<T>::value, {0})); in GetNextInternal()
148 out_tensors->push_back(dense_shape_); in GetNextInternal()
|
| D | map_dataset_op.cc | 60 std::vector<Tensor>* out_tensors) { in MakeDataset() argument
61 return inst_captured_func->Run(ctx, std::move(args), out_tensors); in MakeDataset()
70 std::vector<Tensor>* out_tensors) { in MakeDataset() argument
77 out_tensors->push_back(std::move(args[indices[i]])); in MakeDataset()
79 out_tensors->push_back(args[indices[i]]); in MakeDataset()
82 out_tensors->push_back(captured_inputs[indices[i] - num_args]); in MakeDataset()
205 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
219 out_tensors); in GetNextInternal()
|
| D | tensor_dataset_op_test.cc | 164 std::vector<Tensor> out_tensors; in TEST_P() local
166 TF_EXPECT_OK(iterator->GetNext(iterator_context.get(), &out_tensors, in TEST_P()
169 EXPECT_EQ(out_tensors.size(), test_case.expected_outputs.size()); in TEST_P()
170 for (int i = 0; i < out_tensors.size(); ++i) { in TEST_P()
171 if (out_tensors[i].dtype() == DT_VARIANT) { in TEST_P()
174 const Tensor *output = out_tensors[i].scalar<Variant>()().get<Tensor>(); in TEST_P()
179 TF_EXPECT_OK(ExpectEqual(out_tensors[i], test_case.expected_outputs[i])); in TEST_P()
486 std::vector<Tensor> out_tensors; in TEST_P() local
498 TF_EXPECT_OK(iterator->GetNext(iterator_ctx.get(), &out_tensors, in TEST_P()
501 EXPECT_EQ(out_tensors.size(), test_case.expected_outputs.size()); in TEST_P()
[all …]
|
| D | zip_dataset_op.cc | 127 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
134 out_tensors->clear(); in GetNextInternal()
135 out_tensors->reserve(dataset()->output_dtypes().size()); in GetNextInternal()
143 out_tensors->insert(out_tensors->end(), input_tensors.begin(), in GetNextInternal()
147 out_tensors->clear(); in GetNextInternal()
|
| D | filter_by_component_dataset_op.cc | 111 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
125 input_impl_->GetNext(ctx, out_tensors, end_of_sequence)); in GetNextInternal()
133 matched = out_tensors->back().scalar<bool>()(); in GetNextInternal()
134 out_tensors->pop_back(); in GetNextInternal()
137 out_tensors->clear(); in GetNextInternal()
|
| D | tensor_slice_dataset_op_test.cc | 171 std::vector<Tensor> out_tensors; in TEST_F() local
175 TF_EXPECT_OK(iterator->GetNext(iterator_context.get(), &out_tensors, in TEST_F()
177 for (int i = 0; i < out_tensors.size(); ++i) { in TEST_F()
180 if (out_tensors[i].dtype() == DT_VARIANT) { in TEST_F()
184 out_tensors[i].scalar<Variant>()().get<Tensor>(); in TEST_F()
192 out_tensors[i], in TEST_F()
196 out_tensors.clear(); in TEST_F()
566 std::vector<Tensor> out_tensors; in TEST_F() local
570 TF_EXPECT_OK(iterator->GetNext(iterator_context.get(), &out_tensors, in TEST_F()
578 for (int i = 0; i < out_tensors.size(); ++i) { in TEST_F()
[all …]
|
| D | tensor_slice_dataset_op.cc | 130 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
134 out_tensors->clear(); in GetNextInternal()
135 out_tensors->reserve(dataset()->tensors_.size()); in GetNextInternal()
138 out_tensors->emplace_back( in GetNextInternal()
142 batch_util::CopySliceToElement(t, &out_tensors->back(), i_)); in GetNextInternal()
|
| D | sparse_tensor_slice_dataset_op_test.cc | 159 std::vector<Tensor> out_tensors; in TEST_F() local
162 TF_EXPECT_OK(iterator->GetNext(iterator_ctx.get(), &out_tensors, in TEST_F()
166 ExpectEqual(out_tensors[0], expected_outputs[cur_slice].indices)); in TEST_F()
168 ExpectEqual(out_tensors[1], expected_outputs[cur_slice].values)); in TEST_F()
169 TF_EXPECT_OK(ExpectEqual(out_tensors[2], in TEST_F()
474 std::vector<Tensor> out_tensors; in TEST_F() local
478 TF_EXPECT_OK(iterator->GetNext(iterator_ctx.get(), &out_tensors, in TEST_F()
486 for (int i = 0; i < out_tensors.size(); ++i) { in TEST_F()
488 out_tensors[0], expected_outputs[cur_iteration - 1].indices)); in TEST_F()
489 TF_EXPECT_OK(ExpectEqual(out_tensors[1], in TEST_F()
[all …]
|
| /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/bigtable/kernels/ |
| D | bigtable_sample_key_pairs_dataset_op.cc | 167 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
176 out_tensors->emplace_back(ctx->allocator({}), DT_STRING, in GetNextInternal()
178 out_tensors->back().scalar<string>()() = keys_[index_]; in GetNextInternal()
180 out_tensors->emplace_back(ctx->allocator({}), DT_STRING, in GetNextInternal()
182 out_tensors->back().scalar<string>()() = keys_[index_ + 1]; in GetNextInternal()
|
| D | bigtable_lookup_dataset_op.cc | 132 std::vector<Tensor>* out_tensors, in GetNextInternal() argument
166 TF_RETURN_IF_ERROR(ParseRow(ctx, pair.second, out_tensors)); in GetNextInternal()
178 std::vector<Tensor>* out_tensors) { in ParseRow() argument
179 out_tensors->reserve(dataset()->columns_.size() + 1); in ParseRow()
182 out_tensors->emplace_back(std::move(row_key_tensor)); in ParseRow()
208 out_tensors->emplace_back(std::move(col_tensor)); in ParseRow()
|