| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_def_builder.cc | 104 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in SourceOp() local
106 return opts.FinalizeBuilder(&node_builder); in SourceOp()
112 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in UnaryOp() local
114 node_builder.Input(std::move(input)); in UnaryOp()
115 return opts.FinalizeBuilder(&node_builder); in UnaryOp()
121 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in BinaryOp() local
123 node_builder.Input(std::move(a)).Input(std::move(b)); in BinaryOp()
124 return opts.FinalizeBuilder(&node_builder); in BinaryOp()
130 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in TernaryOp() local
132 node_builder.Input(std::move(a)).Input(std::move(b)).Input(std::move(c)); in TernaryOp()
[all …]
|
| D | BUILD | 57 "node_builder.h",
70 "node_builder.h",
83 "node_builder.cc",
108 "node_builder.h",
182 "node_builder.cc",
183 "node_builder.h",
|
| D | graph_partition.cc | 424 NodeBuilder node_builder(node_name, "Enter", g->op_registry()); in AddControlEnter() local
425 node_builder.Input({"dummy", 0, DT_FLOAT}); in AddControlEnter()
426 node_builder.Attr("frame_name", frame_name); in AddControlEnter()
427 node_builder.Attr("parallel_iterations", parallel_iterations); in AddControlEnter()
429 *status = node_builder.Finalize(g, &res_node, /*consume=*/true); in AddControlEnter()
439 NodeBuilder node_builder(node_name, "Merge", g->op_registry()); in AddControlMerge() local
440 node_builder.Input({{in_name1, 0, DT_FLOAT}, {in_name2, 0, DT_FLOAT}}); in AddControlMerge()
442 *status = node_builder.Finalize(g, &res_node, /*consume=*/true); in AddControlMerge()
470 NodeBuilder node_builder(options.GetNameForOp("Const"), "Const", in EmptyConst() local
477 node_builder.Attr("dtype", dt).Attr("value", proto); in EmptyConst()
[all …]
|
| D | Android.bp | 37 "node_builder.cc",
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_unified_experimental_graph.cc | 177 op_->node_builder.Attr(attr_name, s); in SetAttrString()
181 op_->node_builder.Attr(attr_name, static_cast<int64_t>(value)); in SetAttrInt()
185 op_->node_builder.Attr(attr_name, value); in SetAttrFloat()
189 op_->node_builder.Attr(attr_name, value); in SetAttrBool()
198 op_->node_builder.Attr(attr_name, value); in SetAttrType()
208 op_->node_builder.Attr(attr_name, shape); in SetAttrShape()
220 op_->node_builder.Attr(attr_name, func_name); in SetAttrFunctionName()
242 op_->node_builder.Attr(attr_name, v); in SetAttrStringList()
248 op_->node_builder.Attr(attr_name, in SetAttrFloatList()
254 op_->node_builder.Attr( in SetAttrIntList()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | parallel_concat_optimizer.cc | 52 NodeBuilder node_builder(name, op, OpRegistry::Global(), &debug_info); in Run() local
53 node_builder.Device(n->requested_device()); in Run()
56 node_builder.Attr("_class", colo); in Run()
58 return node_builder; in Run()
|
| D | lower_if_op.cc | 70 Status SetColocationAndFinalize(NodeBuilder node_builder, Graph* graph,
139 Status CondBuilder::SetColocationAndFinalize(NodeBuilder node_builder, in SetColocationAndFinalize() argument
143 node_builder = node_builder.Attr(kColocationAttrName, *coloc_attr_); in SetColocationAndFinalize()
145 return node_builder.Finalize(graph, created_node); in SetColocationAndFinalize()
|
| D | accumulate_n_optimizer.cc | 111 NodeBuilder node_builder(name, op, OpRegistry::Global(), &debug_info); in RewriteIntoTempVariable() local
114 node_builder.Device(n->requested_device()); in RewriteIntoTempVariable()
117 node_builder.Attr(kColocationAttrName, colo); in RewriteIntoTempVariable()
119 return node_builder; in RewriteIntoTempVariable()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | cluster_scoping_pass_test.cc | 76 NodeBuilder node_builder(name, "Stage", opts.op_registry()); in BuildStageNode() local
77 node_builder.Input(values); in BuildStageNode()
78 return opts.FinalizeBuilder(&node_builder); in BuildStageNode()
|
| D | encapsulate_subgraphs_pass_test.cc | 348 NodeBuilder node_builder(opts.GetNameForOp("NoOp"), "NoOp", in Sequencer() local
351 .FinalizeBuilder(&node_builder); in Sequencer()
365 NodeBuilder node_builder(opts.GetNameForOp("Const"), "Const", in KnownShapeBase() local
374 .FinalizeBuilder(&node_builder); in KnownShapeBase()
389 NodeBuilder node_builder(absl::StrCat(call_node, "_key_placeholder"), in KeyPlaceholder() local
396 .FinalizeBuilder(&node_builder); in KeyPlaceholder()
408 NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaRecvAtHost"), in RecvAtHost() local
410 node_builder.Input(std::move(key_input)); in RecvAtHost()
416 .FinalizeBuilder(&node_builder); in RecvAtHost()
428 NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaSendFromHost"), in SendFromHost() local
[all …]
|
| D | encapsulate_subgraphs_pass.cc | 985 NodeBuilder node_builder(options.GetNameForOp("KnownShape"), "Const", in AddDummyShapedNode() local
987 node_builder.Attr("dtype", data_type).Attr("value", dummy_proto); in AddDummyShapedNode()
988 Node* node = options.FinalizeBuilder(&node_builder); in AddDummyShapedNode()
|
| D | mark_for_compilation_pass_test.cc | 1867 NodeBuilder node_builder(name, "Stage", opts.op_registry()); in MakeStageNode() local
1868 node_builder.Input(values); in MakeStageNode()
1869 return opts.FinalizeBuilder(&node_builder); in MakeStageNode()
|
| D | extract_outside_compilation_pass.cc | 1330 void SetMaximalSharding(NodeDefBuilder& node_builder) { in SetMaximalSharding() argument
1335 node_builder.Attr("_XlaSharding", sharding.SerializeAsString()); in SetMaximalSharding()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 790 desc->node_builder.Device(device); in TF_SetDevice()
794 desc->node_builder.Input(&input.oper->node, input.index); in TF_AddInput()
804 desc->node_builder.Input(input_list); in TF_AddInputList()
808 desc->node_builder.ControlInput(&input->node); in TF_AddControlInput()
819 desc->node_builder.Attr(attr_name, s); in TF_SetAttrString()
837 desc->node_builder.Attr(attr_name, v); in TF_SetAttrStringList()
843 desc->node_builder.Attr(attr_name, static_cast<int64_t>(value)); in TF_SetAttrInt()
848 desc->node_builder.Attr( in TF_SetAttrIntList()
855 desc->node_builder.Attr(attr_name, value); in TF_SetAttrFloat()
860 desc->node_builder.Attr(attr_name, in TF_SetAttrFloatList()
[all …]
|
| D | c_api_internal.h | 109 : node_builder(node_name, op_type, g->graph.op_registry()), graph(g) {} in TF_OperationDescription()
111 tensorflow::NodeBuilder node_builder; member
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 473 NodeDefBuilder node_builder(info.engine_name, "TRTEngineOp"); in CreateTRTNode() local
474 if (!info.device.empty()) node_builder.Device(info.device); in CreateTRTNode()
482 node_builder.Input(inputs); in CreateTRTNode()
484 node_builder.ControlInput(c); in CreateTRTNode()
491 node_builder.Attr("input_shapes", input_shape_protos) in CreateTRTNode()
509 node_builder.Attr("profile_strategy", in CreateTRTNode()
513 Status status = node_builder.Finalize(&trt_node); in CreateTRTNode()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.cc | 290 NodeBuilder node_builder( in AddDataset() local
299 node_builder.Input(NodeBuilder::NodeOut(inputs_iter->second)); in AddDataset()
308 node_builder.Input(nodeout_inputs); in AddDataset()
315 *output = opts->FinalizeBuilder(&node_builder); in AddDataset()
830 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in AddDatasetOrTensorHelper() local
832 node_builder.Input(std::move(nodes)); in AddDatasetOrTensorHelper()
833 *output = opts.FinalizeBuilder(&node_builder); in AddDatasetOrTensorHelper()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | gpu_model.cc | 85 data::GpuNodeBuilder node_builder(*builder); in Encode() local
86 node_builder.add_gpu_op(op_fb); in Encode()
87 node_builder.add_input_ids(in_ids_fb); in Encode()
88 node_builder.add_output_ids(out_ids_fb); in Encode()
89 node_builder.add_name(name_fb); in Encode()
90 return node_builder.Finish(); in Encode()
|