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())