Python Resourceの例

コード例 #1

ファイル: config_util.py プロジェクト: zzk0/oneflow

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)

コード例 #2

ファイル: c_api_util.py プロジェクト: ncnnnnn/oneflow

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

コード例 #3

def EnvResource():
    resource = oneflow._oneflow_internal.EnvResource()
    return text_format.Parse(resource, resource_util.Resource())

コード例 #4

ファイル: c_api_util.py プロジェクト: liudyboy/oneflow

def EnvResource():
    resource = oneflow_api.EnvResource()
    return text_format.Parse(resource, resource_util.Resource())

コード例 #5

ファイル: c_api_util.py プロジェクト: liudyboy/oneflow

def CurrentResource():
    resource = oneflow_api.CurrentResource()
    return text_format.Parse(resource, resource_util.Resource())

コード例 #6

def EnvResource():
    resource, error = oneflow_api.EnvResource()
    if error.has_error_type():
        raise JobBuildAndInferCfgError(error)
    return text_format.Parse(resource, resource_util.Resource())