| 384 | |
| 385 | template <class T> |
| 386 | inline void SetNodeAttr(const string& key, const T& value, NodeDef* node) { |
| 387 | AttrValue attr_value; |
| 388 | SetAttrValue(value, &attr_value); |
| 389 | auto* attr_map = node->mutable_attr(); |
| 390 | (*attr_map)[key] = attr_value; |
| 391 | } |
| 392 | template <> |
| 393 | inline void SetNodeAttr(const string& key, const Tensor& tensor, |
| 394 | NodeDef* node) { |
no test coverage detected