Searched refs:def_builder_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | node_builder.cc | 43 : def_builder_(name, op_name, op_registry, debug) {} in NodeBuilder()
46 : def_builder_(name, op_def) {} in NodeBuilder()
49 : def_builder_(def_builder) {} in NodeBuilder()
55 def_builder_.Input(src_node->name(), src_index, dt); in Input()
65 def_builder_.Input(src.name, src.index, src.dt); in Input()
81 def_builder_.Input(gtl::ArraySlice<NodeDefBuilder::NodeOut>(srcs)); in Input()
87 def_builder_.ControlInput(src_node->name()); in ControlInput()
95 def_builder_.ControlInput(src_node->name()); in ControlInputs()
101 def_builder_.Device(device_spec); in Device()
111 def_builder_.Attr("_XlaCluster", xla_cluster); in XlaCluster()
[all …]
|
| D | node_builder.h | 129 const string& node_name() const { return def_builder_.node_name(); } in node_name()
130 const OpDef& op_def() const { return def_builder_.op_def(); } in op_def()
150 NodeDefBuilder def_builder_; variable
161 def_builder_.Attr(attr_name, std::forward<T>(value)); in Attr()
168 def_builder_.Attr(attr_name, value); in Attr()
|