| 314 | } |
| 315 | |
| 316 | bool GraphDefBuilderWrapper::HasAttr(const string& name, |
| 317 | const string& attr_name) const { |
| 318 | const OpDef* op_def = nullptr; |
| 319 | Status s = b_->opts().op_registry()->LookUpOpDef(name, &op_def); |
| 320 | if (!s.ok() || op_def == nullptr) { |
| 321 | return false; |
| 322 | } |
| 323 | return HasAttr(op_def, attr_name); |
| 324 | } |
| 325 | |
| 326 | int64 GetAllocatedBytes(const std::vector<Tensor>& element) { |
| 327 | int64 allocated_bytes = 0; |
no test coverage detected