| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph_def_builder.cc | 104 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in SourceOp() local
112 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in UnaryOp() local
121 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in BinaryOp() local
130 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in TernaryOp() local
|
| H A D | graph_partition.cc | 424 NodeBuilder node_builder(node_name, "Enter", g->op_registry()); in AddControlEnter() local
439 NodeBuilder node_builder(node_name, "Merge", g->op_registry()); in AddControlMerge() local
470 NodeBuilder node_builder(options.GetNameForOp("Const"), "Const", in EmptyConst() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | encapsulate_subgraphs_pass_test.cc | 348 NodeBuilder node_builder(opts.GetNameForOp("NoOp"), "NoOp", in Sequencer() local
365 NodeBuilder node_builder(opts.GetNameForOp("Const"), "Const", in KnownShapeBase() local
389 NodeBuilder node_builder(absl::StrCat(call_node, "_key_placeholder"), in KeyPlaceholder() local
408 NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaRecvAtHost"), in RecvAtHost() local
428 NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaSendFromHost"), in SendFromHost() local
461 NodeBuilder node_builder(opts.GetNameForOp("AddN"), "AddNLikeTest", in AddNLike() local
474 NodeBuilder node_builder(opts.GetNameForOp("Retval"), "_Retval", in RetOp() local
622 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST() local
978 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST() local
1189 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST() local
[all …]
|
| H A D | cluster_scoping_pass_test.cc | 76 NodeBuilder node_builder(name, "Stage", opts.op_registry()); in BuildStageNode() local
|
| H A D | encapsulate_subgraphs_pass.cc | 985 NodeBuilder node_builder(options.GetNameForOp("KnownShape"), "Const", in AddDummyShapedNode() local
|
| H A D | extract_outside_compilation_pass.cc | 1330 void SetMaximalSharding(NodeDefBuilder& node_builder) { in SetMaximalSharding()
|
| H A D | mark_for_compilation_pass_test.cc | 1867 NodeBuilder node_builder(name, "Stage", opts.op_registry()); in MakeStageNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | parallel_concat_optimizer.cc | 52 NodeBuilder node_builder(name, op, OpRegistry::Global(), &debug_info); in Run() local
|
| H A D | accumulate_n_optimizer.cc | 111 NodeBuilder node_builder(name, op, OpRegistry::Global(), &debug_info); in RewriteIntoTempVariable() local
|
| H A D | lower_if_op.cc | 139 Status CondBuilder::SetColocationAndFinalize(NodeBuilder node_builder, in SetColocationAndFinalize()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_internal.h | 111 tensorflow::NodeBuilder node_builder; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | dataset.cc | 290 NodeBuilder node_builder( in AddDataset() local
830 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in AddDatasetOrTensorHelper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_graph.cc | 473 NodeDefBuilder node_builder(info.engine_name, "TRTEngineOp"); in CreateTRTNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | gpu_model.cc | 85 data::GpuNodeBuilder node_builder(*builder); in Encode() local
|