| /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 | graph_partition.cc | 398 NodeBuilder node_builder(node_name, "Enter", g->op_registry()); in AddControlEnter() local
399 node_builder.Input({"dummy", 0, DT_FLOAT}); in AddControlEnter()
400 node_builder.Attr("frame_name", frame_name); in AddControlEnter()
401 node_builder.Attr("parallel_iterations", parallel_iterations); in AddControlEnter()
403 *status = node_builder.Finalize(g, &res_node); in AddControlEnter()
413 NodeBuilder node_builder(node_name, "Merge", g->op_registry()); in AddControlMerge() local
414 node_builder.Input({{in_name1, 0, DT_FLOAT}, {in_name2, 0, DT_FLOAT}}); in AddControlMerge()
416 *status = node_builder.Finalize(g, &res_node); in AddControlMerge()
444 NodeBuilder node_builder(options.GetNameForOp("Const"), "Const", in EmptyConst() local
451 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/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 429 auto node_builder = [](StringPiece op) { in StackTensor() local
433 auto make_const = [&node_builder](const Input::Initializer& val, Graph* graph, in StackTensor()
436 return node_builder("Const") in StackTensor()
449 node_builder("Shape").Input(input_node).Finalize(g, &shape_node)); in StackTensor()
457 TF_RETURN_IF_ERROR(node_builder("StridedSlice") in StackTensor()
465 TF_RETURN_IF_ERROR(node_builder("Reshape") in StackTensor()
472 TF_RETURN_IF_ERROR(node_builder("Shape") in StackTensor()
478 node_builder("OnesLike").Input(ones_shape).Finalize(g, &ones)); in StackTensor()
484 TF_RETURN_IF_ERROR(node_builder("Concat") in StackTensor()
490 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/core/common_runtime/ |
| D | parallel_concat_optimizer.cc | 56 NodeBuilder node_builder(name, op, OpRegistry::Global(), &debug_info); in Run() local
57 node_builder.Device(n->requested_device()); in Run()
60 node_builder.Attr("_class", colo); in Run()
62 return node_builder; in Run()
|
| D | accumulate_n_optimizer.cc | 79 NodeBuilder node_builder(name, op, OpRegistry::Global(), &debug_info); in rewriteNode() local
82 node_builder.Device(n->requested_device()); in rewriteNode()
85 node_builder.Attr(kColocationAttrName, colo); in rewriteNode()
87 return node_builder; in rewriteNode()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 360 NodeBuilder node_builder(opts.GetNameForOp("NoOp"), "NoOp", in Sequencer() local
363 .FinalizeBuilder(&node_builder); in Sequencer()
377 NodeBuilder node_builder(opts.GetNameForOp("Const"), "Const", in KnownShapeBase() local
386 .FinalizeBuilder(&node_builder); in KnownShapeBase()
401 NodeBuilder node_builder(absl::StrCat(call_node, "_key_placeholder"), in KeyPlaceholder() local
408 .FinalizeBuilder(&node_builder); in KeyPlaceholder()
418 NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaRecvAtHost"), in RecvAtHost() local
420 node_builder.Input(std::move(key_input)); in RecvAtHost()
426 .FinalizeBuilder(&node_builder); in RecvAtHost()
437 NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaSendFromHost"), in SendFromHost() local
[all …]
|
| D | encapsulate_subgraphs_pass.cc | 1844 NodeBuilder node_builder(options.GetNameForOp("KnownShape"), "Const", in AddDummyShapedNode() local
1846 node_builder.Attr("dtype", data_type).Attr("value", dummy_proto); in AddDummyShapedNode()
1847 Node* node = options.FinalizeBuilder(&node_builder); in AddDummyShapedNode()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 535 NodeDefBuilder node_builder(info.engine_name, "TRTEngineOp"); in CreateTRTNode() local
536 if (!info.device.empty()) node_builder.Device(info.device); in CreateTRTNode()
544 node_builder.Input(inputs); in CreateTRTNode()
546 node_builder.ControlInput(c); in CreateTRTNode()
555 node_builder.Attr("input_shapes", input_shape_protos) in CreateTRTNode()
652 NodeDefBuilder node_builder(StrCat(name, "_Arg"), in RegisterSegmentFunctionToFunctionLibrary() local
655 TF_RETURN_IF_ERROR(node_builder.Attr("T", node->output_type(0)) in RegisterSegmentFunctionToFunctionLibrary()
679 NodeDefBuilder node_builder(StrCat(name, "_Ret"), in RegisterSegmentFunctionToFunctionLibrary() local
688 node_builder.Input(nout); in RegisterSegmentFunctionToFunctionLibrary()
689 TF_RETURN_IF_ERROR(node_builder.Attr("T", node->output_type(0)) in RegisterSegmentFunctionToFunctionLibrary()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 1212 desc->node_builder.Device(device); in TF_SetDevice()
1216 desc->node_builder.Input(&input.oper->node, input.index); in TF_AddInput()
1226 desc->node_builder.Input(input_list); in TF_AddInputList()
1230 desc->node_builder.ControlInput(&input->node); in TF_AddControlInput()
1241 desc->node_builder.Attr(attr_name, s); in TF_SetAttrString()
1259 desc->node_builder.Attr(attr_name, v); in TF_SetAttrStringList()
1267 desc->node_builder.Attr(attr_name, static_cast<tensorflow::int64>(value)); in TF_SetAttrInt()
1274 desc->node_builder.Attr( in TF_SetAttrIntList()
1282 desc->node_builder.Attr(attr_name, value); in TF_SetAttrFloat()
1287 desc->node_builder.Attr(attr_name, in TF_SetAttrFloatList()
[all …]
|
| D | c_api_internal.h | 117 : node_builder(node_name, op_type, g->graph.op_registry()), graph(g) {} in TF_OperationDescription()
119 tensorflow::NodeBuilder node_builder; member
|
| D | c_api_experimental.cc | 9072 desc->node_builder.Attr(attr.first, attr.second); in TFE_AddEagerOpToGraph()
9078 const tensorflow::OpDef& op_def = desc->node_builder.op_def(); in TFE_AddEagerOpToGraph()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.cc | 217 NodeBuilder node_builder(opts->GetNameForOp(type_string), type_string, in AddDataset() local
225 node_builder.Input(NodeBuilder::NodeOut(inputs_iter->second)); in AddDataset()
234 node_builder.Input(nodeout_inputs); in AddDataset()
241 *output = opts->FinalizeBuilder(&node_builder); in AddDataset()
|
| /external/tensorflow/tensorflow/contrib/cmake/ |
| D | tf_core_cpu.cmake | 57 "${tensorflow_source_dir}/tensorflow/core/graph/node_builder.h"
58 "${tensorflow_source_dir}/tensorflow/core/graph/node_builder.cc"
|
| D | tf_core_framework.cmake | 293 "${tensorflow_source_dir}/tensorflow/core/graph/node_builder.h"
294 "${tensorflow_source_dir}/tensorflow/core/graph/node_builder.cc"
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 1430 "graph/node_builder.h",
2683 "graph/node_builder.h",
2786 "graph/node_builder.cc",
2938 "graph/node_builder.h",
|