MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetIntAttr

Function GetIntAttr

tensorflow/lite/toco/import_tensorflow.cc:88–94  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

86}
87
88int64 GetIntAttr(const NodeDef& node, const string& attr_name) {
89 CHECK(HasAttr(node, attr_name)) << attr_name << " not found in:\n"
90 << node.DebugString();
91 const auto& attr = node.attr().at(attr_name);
92 CHECK_EQ(attr.value_case(), AttrValue::kI);
93 return attr.i();
94}
95
96float GetFloatAttr(const NodeDef& node, const string& attr_name) {
97 CHECK(HasAttr(node, attr_name));

Callers 15

ConvertRandomUniformFunction · 0.85
ConvertSplitOperatorFunction · 0.85
ConvertSplitVOperatorFunction · 0.85
ConvertLRNOperatorFunction · 0.85
ConvertConcatOperatorFunction · 0.85

Calls 5

HasAttrFunction · 0.85
attrMethod · 0.80
DebugStringMethod · 0.45
atMethod · 0.45
iMethod · 0.45

Tested by 1

CompareMethod · 0.68