Home
last modified time | relevance | path

Searched refs:nodedef_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dincomplete_nodedef_builder.cc26 nodedef_.set_name(name); in IncompleteNodeDefBuilder()
27 nodedef_.set_op(op); in IncompleteNodeDefBuilder()
28 MergeDebugInfo(debug, &nodedef_); in IncompleteNodeDefBuilder()
33 AddNodeAttr(attr, type, &nodedef_); in AddAttr()
39 AddNodeAttr(attr, val, &nodedef_); in AddAttr()
45 nodedef_.set_device(device); in Device()
51 *n = graph->AddNode(nodedef_, &status); in Build()
Dincomplete_nodedef_builder.h53 NodeDef nodedef_;
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc201 const tensorflow::NodeDef& nodedef() const { return nodedef_; } in nodedef()
232 if (!nodedef_.ParseFromString(v[1].AsString().str())) { in InitializeNodeDef()
233 nodedef_.Clear(); in InitializeNodeDef()
240 tensorflow::OpRegistry::Global()->LookUp(nodedef_.op(), &op_reg_data_)); in InitializeNodeDef()
241 AddDefaultsToNodeDef(op_reg_data_->op_def, &nodedef_); in InitializeNodeDef()
260 for (const auto& attr : nodedef_.attr()) { in BuildEagerOp()
324 tensorflow::NodeDef nodedef_; member in tflite::flex::OpNode