/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::__anon3dfc05720111::ConcatenateDatasetOp::Dataset 71 return input_->output_dtypes(); in output_dtypes()
|
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::__anon9666baa40111::UniqueDatasetOp::Dataset 66 return input_->output_dtypes(); in output_dtypes()
|
D | padded_batch_dataset_op.cc | 79 OP_REQUIRES(ctx, padding_value_t.dtype() == input->output_dtypes()[i], in MakeDataset() 84 DataTypeString(input->output_dtypes()[i]))); in MakeDataset() 128 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon6afc19300111::PaddedBatchDatasetOp::Dataset 129 return input_->output_dtypes(); in output_dtypes() 170 b->BuildAttrValue(output_dtypes(), &output_types); in AsGraphDefInternal() 285 output_dtypes()[component_index], in GetNextInternal()
|
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::__anone199950c0111::DenseToSparseBatchDatasetOp::Dataset 222 DatasetIterator<Dataset<T>>::dataset()->input_->output_dtypes()[0], in GetNextInternal()
|
D | stats_dataset_ops.cc | 65 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon0bf9c36f0111::LatencyStatsDatasetOp::Dataset 66 return input_->output_dtypes(); in output_dtypes() 159 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon0bf9c36f0111::BytesProducedStatsDatasetOp::Dataset 160 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::__anonea912f600111::ZipDatasetOp::Dataset 116 out_tensors->reserve(dataset()->output_dtypes().size()); in GetNextInternal()
|
D | cache_dataset_ops.cc | 55 num_tensors_(input->output_dtypes().size()), in FileDataset() 78 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon785ee0740111::CacheDatasetOp::FileDataset 79 return input_->output_dtypes(); in output_dtypes() 287 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon785ee0740111::CacheDatasetOp::MemoryDataset 288 return input_->output_dtypes(); in output_dtypes()
|
D | take_dataset_op.cc | 64 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon1b2dbf720111::TakeDatasetOp::Dataset 65 return input_->output_dtypes(); in output_dtypes()
|
D | skip_dataset_op.cc | 64 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anond541dce40111::SkipDatasetOp::Dataset 65 return input_->output_dtypes(); in output_dtypes()
|
D | iterator_ops.cc | 84 IteratorResource(const DataTypeVector& output_dtypes, in IteratorResource() argument 96 output_dtypes_(output_dtypes), in IteratorResource() 190 VerifyTypesMatch(output_dtypes_, iterator->output_dtypes())); in set_iterator() 210 const DataTypeVector& output_dtypes() const { return output_dtypes_; } in output_dtypes() function in tensorflow::__anon70b9ebe50111::IteratorResource 521 VerifyTypesMatch(output_dtypes_, resource->output_dtypes())); in VerifyResource() 620 components.reserve(dataset->output_dtypes().size()); in ComputeAsync() 758 VerifyTypesMatch(output_dtypes_, (*iterator)->output_dtypes())); in TryInit() 1004 iterator_resource->output_dtypes())); in Compute()
|
D | scan_dataset_op.cc | 99 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon963ac2520111::ScanDatasetOp::Dataset 175 output_dtypes().size()); in GetNextInternal() 193 if (state_and_output[i].dtype() != output_dtypes()[output_index]) { in GetNextInternal()
|
D | filter_dataset_op.cc | 77 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon35f544d30111::FilterDatasetOp::Dataset 78 return input_->output_dtypes(); in output_dtypes()
|
D | batch_dataset_op.cc | 70 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon5373aaef0111::BatchDatasetOp::Dataset 71 return input_->output_dtypes(); in output_dtypes()
|
D | tensor_queue_dataset_op.cc | 90 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anondf448c0c0111::PrependFromQueueAndPaddedBatchDataset 524 ctx, input->output_dtypes() == output_types_, in MakeDataset() 527 DataTypeVectorString(input->output_dtypes()), in MakeDataset() 561 OP_REQUIRES(ctx, padding_value_t.dtype() == input->output_dtypes()[i], in MakeDataset() 566 DataTypeString(input->output_dtypes()[i]))); in MakeDataset()
|
D | repeat_dataset_op.cc | 65 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon37b066ce0111::RepeatDatasetOp::Dataset 66 return input_->output_dtypes(); in output_dtypes()
|
D | prefetch_dataset_op.cc | 62 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anonfad5a8de0111::PrefetchDatasetOp::Dataset 63 return input_->output_dtypes(); in output_dtypes()
|
D | window_dataset.cc | 35 const DataTypeVector& output_dtypes() const override { return output_types_; } in output_dtypes() function in tensorflow::__anonebb77fc70111::WindowDataset
|
D | tensor_dataset_op.cc | 62 const DataTypeVector& output_dtypes() const override { return dtypes_; } in output_dtypes() function in tensorflow::__anon601612280111::TensorDatasetOp::Dataset
|
D | reader_dataset_ops.cc | 98 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon15fd51f30111::TextLineDatasetOp::Dataset 332 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon15fd51f30111::FixedLengthRecordDatasetOp::Dataset 552 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon15fd51f30111::TFRecordDatasetOp::Dataset
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | categorical_op_test.py | 38 def output_dtypes(self): member in CategoricalTest 104 for output_dtype in self.output_dtypes(): 109 for output_dtype in self.output_dtypes():
|
/external/tensorflow/tensorflow/contrib/lite/python/ |
D | op_hint.py | 287 output_dtypes = [None] * len(call.outputs) 293 output_dtypes[output_index] = tensor.dtype.as_datatype_enum 296 current_graph_def, input_names, output_names, output_dtypes,
|
/external/tensorflow/tensorflow/contrib/data/kernels/ |
D | ignore_errors_dataset_op.cc | 53 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon9eb1baf50111::IgnoreErrorsDatasetOp::Dataset 54 return input_->output_dtypes(); in output_dtypes()
|
/external/tensorflow/tensorflow/contrib/framework/python/framework/ |
D | graph_util.py | 37 def fuse_op(graph_def, input_nodes, output_nodes, output_dtypes, argument 110 new_node.attr["_output_types"].list.type[:] = output_dtypes
|
/external/tensorflow/tensorflow/contrib/framework/python/ops/ |
D | critical_section_ops.py | 189 flat_dtypes = nest.flatten(fn_op.output_dtypes) 247 else nest.pack_sequence_as(fn_op.output_dtypes, flat_outputs))
|
/external/tensorflow/tensorflow/core/framework/ |
D | dataset.h | 339 virtual const DataTypeVector& output_dtypes() const = 0; 407 virtual const DataTypeVector& output_dtypes() const = 0; 506 const DataTypeVector& output_dtypes() const override { in output_dtypes() function 507 return params_.dataset->output_dtypes(); in output_dtypes()
|