def _set_attr_to_resource(attr_name, attr_value):
sess = session_ctx.GetDefaultSession()
if sess.status_ == sess.Status.INITED:
reso_config = resource_util.Resource()
setattr(reso_config, attr_name, attr_value)
sess.update_resource_eagerly(reso_config)
else:
setattr(sess.config_proto.resource, attr_name, attr_value)
def EnvResource():
resource, error_str = oneflow_internal.EnvResource()
error = text_format.Parse(error_str, error_util.ErrorProto())
if error.HasField("error_type"):
raise JobBuildAndInferError(error)
return text_format.Parse(resource, resource_util.Resource())
def EnvResource():
resource = oneflow._oneflow_internal.EnvResource()
return text_format.Parse(resource, resource_util.Resource())
def EnvResource():
resource = oneflow_api.EnvResource()
return text_format.Parse(resource, resource_util.Resource())
def CurrentResource():
resource = oneflow_api.CurrentResource()
return text_format.Parse(resource, resource_util.Resource())
def EnvResource():
resource, error = oneflow_api.EnvResource()
if error.has_error_type():
raise JobBuildAndInferCfgError(error)
return text_format.Parse(resource, resource_util.Resource())
