| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | directed_interleave_dataset_op.cc | 39 selector_input->output_dtypes().size() == 1 && in MakeDataset()
40 selector_input->output_dtypes()[0] == DT_INT64 && in MakeDataset()
54 ctx, data_inputs[0]->output_dtypes() == input->output_dtypes(), in MakeDataset()
58 DataTypeVectorString(data_inputs[0]->output_dtypes()), in MakeDataset()
60 DataTypeVectorString(input->output_dtypes()))); in MakeDataset()
100 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon9a3736c80111::DirectedInterleaveDatasetOp::Dataset
101 return data_inputs_[0]->output_dtypes(); in output_dtypes()
|
| D | indexed_dataset_op.cc | 49 virtual const DataTypeVector& output_dtypes() const = 0;
104 const DataTypeVector& output_dtypes, in MaterializedDatasetResource() argument
106 : output_dtypes_(output_dtypes), output_shapes_(output_shapes) {} in MaterializedDatasetResource()
127 const DataTypeVector& output_dtypes() const { return output_dtypes_; } in output_dtypes() function in tensorflow::data::__anon0a1add170111::MaterializedDatasetResource
136 VerifyTypesMatch(output_dtypes_, dataset->output_dtypes())); in set_materialized_dataset()
312 VerifyTypesMatch(output_dtypes_, resource->output_dtypes())); in VerifyResource()
378 auto expected_types = materialized_resource->output_dtypes(); in Compute()
430 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon0a1add170111::IdentityIndexedDatasetOp::Dataset
501 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon0a1add170111::IdentityIndexedDatasetOp::Dataset::Materialized
502 return dataset_->output_dtypes(); in output_dtypes()
|
| D | dense_to_sparse_batch_dataset_op.cc | 37 ctx, input->output_dtypes().size() == 1, in MakeDataset()
65 switch (input->output_dtypes()[0]) { in MakeDataset()
72 input->output_dtypes()[0])); in MakeDataset()
103 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon017c0f150111::DenseToSparseBatchDatasetOp::Dataset
104 static DataTypeVector* output_dtypes = new DataTypeVector({DT_VARIANT}); in output_dtypes() local
105 return *output_dtypes; in output_dtypes()
235 DatasetIterator<Dataset<T>>::dataset()->input_->output_dtypes()[0], in GetNextInternal()
|
| D | unique_dataset_op.cc | 34 OP_REQUIRES(ctx, input->output_dtypes().size() == 1, in MakeDataset()
38 DataType input_dtype = input->output_dtypes()[0]; in MakeDataset()
65 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon1fc143cd0111::UniqueDatasetOp::Dataset
66 return input_->output_dtypes(); in output_dtypes()
|
| D | threadpool_dataset_op.cc | 162 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon808a52080111::ThreadPoolDatasetOp::Dataset
163 return input_->output_dtypes(); in output_dtypes()
269 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon808a52080111::MaxIntraOpParallelismDatasetOp::Dataset
270 return input_->output_dtypes(); in output_dtypes()
369 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon808a52080111::PrivateThreadPoolDatasetOp::Dataset
370 return input_->output_dtypes(); in output_dtypes()
|
| D | stats_dataset_ops.cc | 67 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon95544c980111::LatencyStatsDatasetOp::Dataset
68 return input_->output_dtypes(); in output_dtypes()
177 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon95544c980111::BytesProducedStatsDatasetOp::Dataset
178 return input_->output_dtypes(); in output_dtypes()
|
| D | scan_dataset_op.cc | 91 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anona5deed3b0111::ScanDatasetOp::Dataset
185 output_dtypes().size()); in GetNextInternal()
190 dataset()->state_types_.size() + output_dtypes().size()); in GetNextInternal()
205 if (state_and_output[i].dtype() != output_dtypes()[output_index]) { in GetNextInternal()
|
| D | parse_example_dataset_op.cc | 195 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon4380089c0111::ParseExampleDatasetOp::Dataset
294 dataset_->output_dtypes()[output_index]) in MapFunc()
297 << DataTypeString(dataset_->output_dtypes()[output_index]) in MapFunc()
322 dataset_->output_dtypes()[output_index]) in MapFunc()
325 << DataTypeString(dataset_->output_dtypes()[output_index]) in MapFunc()
|
| D | unbatch_dataset_op.cc | 65 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anone5f5295b0111::UnbatchDatasetOp::Dataset
66 return input_->output_dtypes(); in output_dtypes()
198 tensors_.resize(dataset()->output_dtypes().size()); in RestoreInternal()
|
| D | sleep_dataset_op.cc | 61 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anone8f2f06f0111::SleepDatasetOp::Dataset
62 return input_->output_dtypes(); in output_dtypes()
|
| D | ignore_errors_dataset_op.cc | 52 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anonc0c267360111::IgnoreErrorsDatasetOp::Dataset
53 return input_->output_dtypes(); in output_dtypes()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op.cc | 32 OP_REQUIRES(ctx, input->output_dtypes() == to_concatenate->output_dtypes(), in MakeDataset()
36 (DataTypeVectorString(input->output_dtypes()), in MakeDataset()
37 DataTypeVectorString(to_concatenate->output_dtypes())))); in MakeDataset()
70 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anondcc1f7fe0111::ConcatenateDatasetOp::Dataset
71 return input_->output_dtypes(); in output_dtypes()
|
| D | padded_batch_dataset_op.cc | 86 OP_REQUIRES(ctx, padding_value_t.dtype() == input->output_dtypes()[i], in MakeDataset()
91 DataTypeString(input->output_dtypes()[i]))); in MakeDataset()
142 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anone2805d3c0111::PaddedBatchDatasetOp::Dataset
143 return input_->output_dtypes(); in output_dtypes()
197 b->BuildAttrValue(output_dtypes(), &output_types); in AsGraphDefInternal()
321 output_dtypes()[component_index], in GetNextInternal()
|
| D | take_dataset_op.h | 45 const DataTypeVector& output_dtypes() const override { in output_dtypes() function
46 return input_->output_dtypes(); in output_dtypes()
|
| D | zip_dataset_op.cc | 48 for (DataType dt : input->output_dtypes()) { in Dataset()
69 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon8a8355ec0111::ZipDatasetOp::Dataset
135 out_tensors->reserve(dataset()->output_dtypes().size()); in GetNextInternal()
|
| D | window_dataset_op.cc | 85 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon256b48110111::WindowDatasetOp::Dataset
86 static DataTypeVector* output_dtypes = new DataTypeVector({DT_VARIANT}); in output_dtypes() local
87 return *output_dtypes; in output_dtypes()
237 {dataset()->input_->output_dtypes()[idx]}); in GetNextInternal()
|
| D | tensor_dataset_op.cc | 41 VerifyTypesMatch((*output)->output_dtypes(), output_types_)); in MakeDataset()
63 const DataTypeVector& output_dtypes() const override { return dtypes_; } in output_dtypes() function in tensorflow::data::__anon67eb3d340111::TensorDatasetOp::Dataset
|
| D | iterator_ops.cc | 56 IteratorResource(Env* env, const DataTypeVector& output_dtypes, in IteratorResource() argument
67 output_dtypes_(output_dtypes), in IteratorResource()
175 VerifyTypesMatch(output_dtypes_, new_state->iterator->output_dtypes())); in Restore()
247 VerifyTypesMatch(output_dtypes_, iterator->output_dtypes())); in SetIteratorFromDataset()
259 const DataTypeVector& output_dtypes() const { return output_dtypes_; } in output_dtypes() function in tensorflow::data::IteratorResource
470 VerifyTypesMatch(output_dtypes_, resource->output_dtypes())); in VerifyResource()
601 components.reserve(dataset->output_dtypes().size()); in ComputeAsync()
861 VerifyTypesMatch(output_dtypes_, (*iterator)->output_dtypes())); in TryInit()
1107 iterator_resource->output_dtypes())); in Compute()
|
| D | shard_dataset_op.cc | 74 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::__anon3d7918cb0111::ShardDatasetOp::Dataset
75 return input_->output_dtypes(); in output_dtypes()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | map_defun.py | 28 output_dtypes, argument
55 if not isinstance(output_dtypes, list):
67 output_dtypes, output_shapes, concrete_fn,
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | math_test.py | 101 output_dtypes = []
115 del output_dtypes[:]
116 output_dtypes.extend([t.dtype for t in outputs])
121 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=output_dtypes)
195 output_dtypes = []
201 del output_dtypes[:]
202 output_dtypes.extend([t.dtype for t in outputs])
206 self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=output_dtypes)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | categorical_op_test.py | 39 def output_dtypes(self): member in CategoricalTest
105 for output_dtype in self.output_dtypes():
110 for output_dtype in self.output_dtypes():
144 for output_dtype in self.output_dtypes():
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 225 Status GetOutputDTypes(EagerOperation* op, DataTypeVector* output_dtypes) { in GetOutputDTypes() argument
237 TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, *op_def, output_dtypes)); in GetOutputDTypes()
464 const DataTypeVector& output_dtypes = kernel->output_dtypes(); in EagerLocalExecute() local
465 const int output_dtypes_size = static_cast<int>(output_dtypes.size()); in EagerLocalExecute()
467 return errors::InvalidArgument("Expecting ", output_dtypes.size(), in EagerLocalExecute()
510 output_dtypes[i], ctx); in EagerLocalExecute()
514 graph_collector, output_dtypes, *retvals); in EagerLocalExecute()
697 DataTypeVector output_dtypes; in EagerRemoteExecute()
698 TF_RETURN_IF_ERROR(GetOutputDTypes(op, &output_dtypes)); in EagerRemoteExecute()
700 if (*num_retvals != output_dtypes.size()) { in EagerRemoteExecute()
[all …]
|
| D | kernel_and_device.h | 109 virtual const DataTypeVector& output_dtypes() const = 0;
163 const DataTypeVector& output_dtypes() const override { in output_dtypes() function
224 const DataTypeVector& output_dtypes() const override { in output_dtypes() function
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | graph_util.py | 37 def fuse_op(graph_def, input_nodes, output_nodes, output_dtypes, argument
115 new_node.attr["_output_types"].list.type[:] = output_dtypes
|