Home
last modified time | relevance | path

Searched refs:AddDataset (Results 1 – 25 of 60) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/data/
Dmodel_dataset_op.cc101 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
110 b->AddDataset(this, {input_graph_node}, in AsGraphDefInternal()
Dshuffle_dataset_op.cc479 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
535 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
598 b->AddDataset(this, in AsGraphDefInternal()
787 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
849 b->AddDataset(this, in AsGraphDefInternal()
Dtensor_dataset_op.cc92 TF_RETURN_IF_ERROR(b->AddDataset(this, {}, {{0, components}}, in AsGraphDefInternal()
Dcache_dataset_ops.cc629 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph, filename}, output)); in AsGraphDefInternal()
652 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
959 b->AddDataset(this, {input_node, filename_node}, output)); in AsGraphDefInternal()
1007 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
Dskip_dataset_op.cc95 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
Dtake_dataset_op.cc202 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node, count}, output)); in AsGraphDefInternal()
Dtensor_slice_dataset_op.cc105 TF_RETURN_IF_ERROR(b->AddDataset(this, {}, {{0, components}}, in AsGraphDefInternal()
Dzip_dataset_op.cc115 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
Dconcatenate_dataset_op.cc109 b->AddDataset(this, {input_graph, to_concatenate_graph}, output)); in AsGraphDefInternal()
Dmap_dataset_op.cc122 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dstats_dataset_ops.cc99 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_node, tag_node}, output)); in AsGraphDefInternal()
216 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_node, tag_node}, output)); in AsGraphDefInternal()
Dthreadpool_dataset_op.cc194 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
326 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
451 b->AddDataset(this, {input_graph_node, num_threads_node}, output)); in AsGraphDefInternal()
Dignore_errors_dataset_op.cc88 b->AddDataset(this, {std::make_pair(0, input_graph_node)}, {}, in AsGraphDefInternal()
Dsleep_dataset_op.cc92 return b->AddDataset(this, in AsGraphDefInternal()
Drandom_dataset_op.cc81 TF_RETURN_IF_ERROR(b->AddDataset(this, {seed, seed2}, output)); in AsGraphDefInternal()
Dassert_cardinality_dataset_op.cc89 b->AddDataset(this, {input_graph_node, cardinality_node}, output)); in AsGraphDefInternal()
Dassert_next_dataset_op.cc85 b->AddDataset(this, {input_graph_node, transformations_node}, output)); in AsGraphDefInternal()
Dtake_while_dataset_op.cc113 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
Dunique_dataset_op.cc72 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_graph_node}, output)); in AsGraphDefInternal()
Dsql_dataset_op.cc124 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
Drebatch_dataset_op.cc107 b->AddDataset(this, {input_graph_node, num_replicas}, output)); in AsGraphDefInternal()
363 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
Dset_stats_aggregator_dataset_op.cc163 TF_RETURN_IF_ERROR(b->AddDataset( in AsGraphDefInternal()
Dsampling_dataset_op.cc95 b->AddDataset(this, {input_graph_node, rate, seed, seed2}, output)); in AsGraphDefInternal()
Dlmdb_dataset_op.cc69 TF_RETURN_IF_ERROR(b->AddDataset(this, {filenames}, output)); in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h214 Status AddDataset(const DatasetBase* dataset, in AddDataset() function
216 return AddDataset(dataset, inputs, {}, output); in AddDataset()
227 Status AddDataset(const DatasetBase* dataset, in AddDataset() function
235 return AddDataset(dataset, enumerated_inputs, {}, attrs, output); in AddDataset()
238 Status AddDataset(

123