| 101 | } |
| 102 | |
| 103 | bool GetBoolAttr(const NodeDef& node, const string& attr_name) { |
| 104 | CHECK(HasAttr(node, attr_name)); |
| 105 | const auto& attr = node.attr().at(attr_name); |
| 106 | CHECK_EQ(attr.value_case(), AttrValue::kB); |
| 107 | return attr.b(); |
| 108 | } |
| 109 | |
| 110 | tensorflow::DataType GetDataTypeAttr(const NodeDef& node, |
| 111 | const string& attr_name) { |
no test coverage detected