Home
last modified time | relevance | path

Searched refs:output_dtypes (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Dconcatenate_dataset_op.cc32 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()
Dunique_dataset_op.cc34 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()
Dpadded_batch_dataset_op.cc79 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()
Ddense_to_sparse_batch_dataset_op.cc37 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()
Dstats_dataset_ops.cc65 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()
Dzip_dataset_op.cc48 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()
Dcache_dataset_ops.cc55 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()
Dtake_dataset_op.cc64 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon1b2dbf720111::TakeDatasetOp::Dataset
65 return input_->output_dtypes(); in output_dtypes()
Dskip_dataset_op.cc64 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anond541dce40111::SkipDatasetOp::Dataset
65 return input_->output_dtypes(); in output_dtypes()
Diterator_ops.cc84 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()
Dscan_dataset_op.cc99 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()
Dfilter_dataset_op.cc77 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon35f544d30111::FilterDatasetOp::Dataset
78 return input_->output_dtypes(); in output_dtypes()
Dbatch_dataset_op.cc70 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon5373aaef0111::BatchDatasetOp::Dataset
71 return input_->output_dtypes(); in output_dtypes()
Dtensor_queue_dataset_op.cc90 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()
Drepeat_dataset_op.cc65 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anon37b066ce0111::RepeatDatasetOp::Dataset
66 return input_->output_dtypes(); in output_dtypes()
Dprefetch_dataset_op.cc62 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::__anonfad5a8de0111::PrefetchDatasetOp::Dataset
63 return input_->output_dtypes(); in output_dtypes()
Dwindow_dataset.cc35 const DataTypeVector& output_dtypes() const override { return output_types_; } in output_dtypes() function in tensorflow::__anonebb77fc70111::WindowDataset
Dtensor_dataset_op.cc62 const DataTypeVector& output_dtypes() const override { return dtypes_; } in output_dtypes() function in tensorflow::__anon601612280111::TensorDatasetOp::Dataset
Dreader_dataset_ops.cc98 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/
Dcategorical_op_test.py38 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/
Dop_hint.py287 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/
Dignore_errors_dataset_op.cc53 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/
Dgraph_util.py37 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/
Dcritical_section_ops.py189 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/
Ddataset.h339 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()

12