/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
H A D | group_by_reducer_dataset_op.cc | 134 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 D | group_by_window_dataset_op.cc | 141 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 D | take_while_dataset_op.cc | 105 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
H A D | io_ops.cc | 271 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 D | scan_dataset_op.cc | 151 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
H A D | choose_fastest_branch_dataset_op.cc | 290 TF_RETURN_IF_ERROR(captured_func->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
H A D | map_and_batch_dataset_op.cc | 162 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
H A D | snapshot_dataset_op.cc | 186 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 D | parallel_interleave_dataset_op.cc | 176 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
H A D | data_service_dataset_op.cc | 387 TF_RETURN_IF_ERROR(captured_uncompress_func_->AddToGraph( in AsGraphDefInternal()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
H A D | captured_function.h | 169 Status AddToGraph(SerializationContext* ctx,
|
H A D | captured_function.cc | 503 Status CapturedFunction::AddToGraph( in AddToGraph() function in tensorflow::data::CapturedFunction
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | filter_dataset_op.cc | 96 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
H A D | map_dataset_op.cc | 124 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
H A D | flat_map_dataset_op.cc | 106 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
H A D | interleave_dataset_op.cc | 112 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
H A D | parallel_filter_dataset_op.cc | 112 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
H A D | parallel_map_dataset_op.cc | 178 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|
H A D | parallel_interleave_dataset_op.cc | 292 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
|