Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A 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()
H A 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()
H A Dtake_while_dataset_op.cc105 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
H A Dio_ops.cc271 TF_RETURN_IF_ERROR(shard_func_->AddToGraph(ctx, b, &shard_func_other_args, in AsGraphDefInternal()
607 TF_RETURN_IF_ERROR(captured_func_->AddToGraph( in AsGraphDefInternal()
H A Dscan_dataset_op.cc151 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
H A Dchoose_fastest_branch_dataset_op.cc290 TF_RETURN_IF_ERROR(captured_func->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
H A Dmap_and_batch_dataset_op.cc162 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
H A Dsnapshot_dataset_op.cc186 TF_RETURN_IF_ERROR(reader_func_->AddToGraph(ctx, b, &reader_func_other_args, in AsGraphDefInternal()
191 TF_RETURN_IF_ERROR(shard_func_->AddToGraph(ctx, b, &shard_func_other_args, in AsGraphDefInternal()
H A Dparallel_interleave_dataset_op.cc176 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
H A Ddata_service_dataset_op.cc387 TF_RETURN_IF_ERROR(captured_uncompress_func_->AddToGraph( in AsGraphDefInternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dcaptured_function.h169 Status AddToGraph(SerializationContext* ctx,
H A Dcaptured_function.cc503 Status CapturedFunction::AddToGraph( in AddToGraph() function in tensorflow::data::CapturedFunction
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dfilter_dataset_op.cc96 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
H A Dmap_dataset_op.cc124 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
H A Dflat_map_dataset_op.cc106 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
H A Dinterleave_dataset_op.cc112 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
H A Dparallel_filter_dataset_op.cc112 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
H A Dparallel_map_dataset_op.cc178 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
H A Dparallel_interleave_dataset_op.cc292 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()