Home
last modified time | relevance | path

Searched refs:AddToGraph (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc134 captured_key_func_->AddToGraph(ctx, b, &key_func_other_arguments_node, in AsGraphDefInternal()
139 TF_RETURN_IF_ERROR(captured_init_func_->AddToGraph( in AsGraphDefInternal()
145 TF_RETURN_IF_ERROR(captured_reduce_func_->AddToGraph( in AsGraphDefInternal()
151 TF_RETURN_IF_ERROR(captured_finalize_func_->AddToGraph( in AsGraphDefInternal()
Dgroup_by_window_dataset_op.cc141 captured_key_func_->AddToGraph(ctx, b, &key_func_other_arguments_node, in AsGraphDefInternal()
146 TF_RETURN_IF_ERROR(captured_reduce_func_->AddToGraph( in AsGraphDefInternal()
152 TF_RETURN_IF_ERROR(captured_window_size_func_->AddToGraph( in AsGraphDefInternal()
Dtake_while_dataset_op.cc105 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
Dscan_dataset_op.cc143 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
Dio_ops.cc224 TF_RETURN_IF_ERROR(captured_func_->AddToGraph( in AsGraphDefInternal()
Dchoose_fastest_branch_dataset_op.cc290 TF_RETURN_IF_ERROR(captured_func->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
Dmap_and_batch_dataset_op.cc162 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
Dparallel_interleave_dataset_op.cc174 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
Dsnapshot_dataset_op.cc371 TF_RETURN_IF_ERROR(reader_func_->AddToGraph(ctx, b, &reader_func_other_args, in AsGraphDefInternal()
376 TF_RETURN_IF_ERROR(shard_func_->AddToGraph(ctx, b, &shard_func_other_args, in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.h148 Status AddToGraph(SerializationContext* ctx,
Dmap_dataset_op.cc102 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
Dfilter_dataset_op.cc96 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
Dflat_map_dataset_op.cc96 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
Dinterleave_dataset_op.cc112 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
Dparallel_map_dataset_op.cc144 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
Dcaptured_function.cc565 Status CapturedFunction::AddToGraph( in AddToGraph() function in tensorflow::data::CapturedFunction
Dparallel_interleave_dataset_op.cc254 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()