| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_def_builder.cc | 103 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in SourceOp() local
105 return opts.FinalizeBuilder(&node_builder); in SourceOp()
111 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in UnaryOp() local
113 node_builder.Input(std::move(input)); in UnaryOp()
114 return opts.FinalizeBuilder(&node_builder); in UnaryOp()
120 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in BinaryOp() local
122 node_builder.Input(std::move(a)).Input(std::move(b)); in BinaryOp()
123 return opts.FinalizeBuilder(&node_builder); in BinaryOp()
|
| D | BUILD | 52 "node_builder.h",
65 "node_builder.h",
78 "node_builder.cc",
103 "node_builder.h",
177 "node_builder.cc",
178 "node_builder.h",
|
| D | graph_partition.cc | 425 NodeBuilder node_builder(node_name, "Enter", g->op_registry()); in AddControlEnter() local
426 node_builder.Input({"dummy", 0, DT_FLOAT}); in AddControlEnter()
427 node_builder.Attr("frame_name", frame_name); in AddControlEnter()
428 node_builder.Attr("parallel_iterations", parallel_iterations); in AddControlEnter()
430 *status = node_builder.Finalize(g, &res_node, /*consume=*/true); in AddControlEnter()
440 NodeBuilder node_builder(node_name, "Merge", g->op_registry()); in AddControlMerge() local
441 node_builder.Input({{in_name1, 0, DT_FLOAT}, {in_name2, 0, DT_FLOAT}}); in AddControlMerge()
443 *status = node_builder.Finalize(g, &res_node, /*consume=*/true); in AddControlMerge()
471 NodeBuilder node_builder(options.GetNameForOp("Const"), "Const", in EmptyConst() local
478 node_builder.Attr("dtype", dt).Attr("value", proto); in EmptyConst()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | decode_csv_vectorizer.cc | 40 auto node_builder = NodeBuilder(node.type_string(), node.type_string()) in Vectorize() local
45 node_builder = node_builder.Attr(attr.first, attr.second); in Vectorize()
47 TF_RETURN_IF_ERROR(node_builder.Finalize(outer_scope, &new_node)); in Vectorize()
|
| D | parse_single_example_vectorizer.cc | 72 auto node_builder = in Vectorize() local
84 node_builder = node_builder.Attr(attr, *val); in Vectorize()
87 TF_RETURN_IF_ERROR(node_builder.Finalize(outer_scope, &new_node)); in Vectorize()
|
| D | cwise_op_vectorizer.cc | 120 auto node_builder = NodeBuilder(strings::StrCat("vectorized/", node.name()), in CwiseVectorizeHelper() local
123 node_builder = node_builder.Input(input.node, input.output_index); in CwiseVectorizeHelper()
126 node_builder = node_builder.Attr(attr_slice.first, attr_slice.second); in CwiseVectorizeHelper()
128 TF_RETURN_IF_ERROR(node_builder.Finalize(outer_scope, &new_node)); in CwiseVectorizeHelper()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_unified_experimental_graph.cc | 176 op_->node_builder.Attr(attr_name, s); in SetAttrString()
182 op_->node_builder.Attr(attr_name, static_cast<tensorflow::int64>(value)); in SetAttrInt()
186 op_->node_builder.Attr(attr_name, value); in SetAttrFloat()
190 op_->node_builder.Attr(attr_name, value); in SetAttrBool()
199 op_->node_builder.Attr(attr_name, value); in SetAttrType()
211 op_->node_builder.Attr(attr_name, shape); in SetAttrShape()
223 op_->node_builder.Attr(attr_name, func_name); in SetAttrFunctionName()
245 op_->node_builder.Attr(attr_name, v); in SetAttrStringList()
251 op_->node_builder.Attr(attr_name, in SetAttrFloatList()
259 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/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 422 auto node_builder = [](StringPiece op) { in StackTensor() local
426 auto make_const = [&node_builder](const Input::Initializer& val, Graph* graph, in StackTensor()
429 return node_builder("Const") in StackTensor()
442 node_builder("Shape").Input(input_node).Finalize(g, &shape_node)); in StackTensor()
450 TF_RETURN_IF_ERROR(node_builder("StridedSlice") in StackTensor()
458 TF_RETURN_IF_ERROR(node_builder("Reshape") in StackTensor()
465 TF_RETURN_IF_ERROR(node_builder("Shape") in StackTensor()
471 node_builder("OnesLike").Input(ones_shape).Finalize(g, &ones)); in StackTensor()
477 TF_RETURN_IF_ERROR(node_builder("Concat") in StackTensor()
483 TF_RETURN_IF_ERROR(node_builder("ExpandDims") in StackTensor()
[all …]
|
| D | vectorization_utils_test.cc | 895 auto node_builder = NodeBuilder("op", op_type); in CwiseTestHelper() local
903 node_builder = node_builder.Input(arg); in CwiseTestHelper()
905 TF_RETURN_IF_ERROR(node_builder.Finalize(&graph, &op)); in CwiseTestHelper()
|
| /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 | 1011 NodeBuilder node_builder(options.GetNameForOp("KnownShape"), "Const", in AddDummyShapedNode() local
1013 node_builder.Attr("dtype", data_type).Attr("value", dummy_proto); in AddDummyShapedNode()
1014 Node* node = options.FinalizeBuilder(&node_builder); in AddDummyShapedNode()
|
| D | mark_for_compilation_pass_test.cc | 1736 NodeBuilder node_builder(name, "Stage", opts.op_registry()); in MakeStageNode() local
1737 node_builder.Input(values); in MakeStageNode()
1738 return opts.FinalizeBuilder(&node_builder); in MakeStageNode()
|
| /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()
845 desc->node_builder.Attr(attr_name, static_cast<tensorflow::int64>(value)); in TF_SetAttrInt()
852 desc->node_builder.Attr( in TF_SetAttrIntList()
860 desc->node_builder.Attr(attr_name, value); in TF_SetAttrFloat()
865 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/core/framework/ |
| D | dataset.cc | 240 NodeBuilder node_builder(opts->GetNameForOp(type_string), type_string, in AddDataset() local
248 node_builder.Input(NodeBuilder::NodeOut(inputs_iter->second)); in AddDataset()
257 node_builder.Input(nodeout_inputs); in AddDataset()
264 *output = opts->FinalizeBuilder(&node_builder); in AddDataset()
519 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in AddDatasetOrTensorHelper() local
521 node_builder.Input(std::move(nodes)); in AddDatasetOrTensorHelper()
522 *output = opts.FinalizeBuilder(&node_builder); in AddDatasetOrTensorHelper()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 467 NodeDefBuilder node_builder(info.engine_name, "TRTEngineOp"); in CreateTRTNode() local
468 if (!info.device.empty()) node_builder.Device(info.device); in CreateTRTNode()
476 node_builder.Input(inputs); in CreateTRTNode()
478 node_builder.ControlInput(c); in CreateTRTNode()
485 node_builder.Attr("input_shapes", input_shape_protos) in CreateTRTNode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | serialization.cc | 930 data::CLNodeBuilder node_builder(*builder); in Encode() local
931 node_builder.add_gpu_op(op_fb); in Encode()
932 node_builder.add_input_ids(in_ids_fb); in Encode()
933 node_builder.add_output_ids(out_ids_fb); in Encode()
934 node_builder.add_name(name_fb); in Encode()
935 return node_builder.Finish(); in Encode()
|