forward_op_->attrs.Set(attr_name, value);
  return op_->SetAttrBool(attr_name, value);
}
absl::Status SetAttrType(AbstractOperation* op_, const char* attr_name,
                         DataType value, ForwardOperation* forward_op_) {
  forward_op_->attrs.Set(attr_name, value);
  return op_->SetAttrType(attr_name, value);
}
absl::Status SetAttrShape(AbstractOperation* op_, const char* attr_name,