Beispiel #1
0
def RegisterMethod4Class():
    eager_blob_util.RegisterMethod4EagerPhysicalBlob()

    blob_trait.RegisterBlobOperatorTraitMethod(oneflow_api.EagerPhysicalBlob)
    blob_trait.RegisterBlobOperatorTraitMethod(oneflow_api.ConsistentBlob)
    blob_trait.RegisterBlobOperatorTraitMethod(oneflow_api.MirroredBlob)

    remote_blob_util.RegisterMethod4EagerBlobTrait()
    remote_blob_util.RegisterMethod4LazyConsistentBlob()
    remote_blob_util.RegisterMethod4LazyMirroredBlob()
    remote_blob_util.RegisterMethod4EagerConsistentBlob()
    blob_cache_util.RegisterMethodAndAttr4BlobCache()
def RegisterMethod4Class():
    eager_blob_util.RegisterMethod4EagerPhysicalBlob()

    blob_trait.RegisterBlobOperatorTraitMethod(oneflow_api.EagerPhysicalBlob)
    blob_trait.RegisterBlobOperatorTraitMethod(oneflow_api.ConsistentBlob)
    blob_trait.RegisterBlobOperatorTraitMethod(oneflow_api.MirroredBlob)

    remote_blob_util.RegisterMethod4EagerBlobTrait()
    remote_blob_util.RegisterMethod4LazyConsistentBlob()
    remote_blob_util.RegisterMethod4LazyMirroredBlob()
    remote_blob_util.RegisterMethod4EagerConsistentBlob()

    vm_util.RegisterMethod4InstructionsBuilder()
def RegisterMethod4Class():
    op_expr_util.RegisterMethod4UserOpExpr()

    eager_blob_util.RegisterMethod4EagerPhysicalBlob()

    blob_trait.RegisterBlobOperatorTraitMethod(oneflow_api.EagerPhysicalBlob)
    blob_trait.RegisterBlobOperatorTraitMethod(oneflow_api.ConsistentBlob)
    blob_trait.RegisterBlobOperatorTraitMethod(oneflow_api.MirroredBlob)

    remote_blob_util.RegisterMethod4EagerBlobTrait()
    remote_blob_util.RegisterMethod4LazyConsistentBlob()
    remote_blob_util.RegisterMethod4LazyMirroredBlob()
    remote_blob_util.RegisterMethod4EagerConsistentBlob()
def RegisterMethod4Class():
    op_expr_util.RegisterMethod4UserOpExpr()
    functional.RegisterFunctionalApis()

    eager_blob_util.RegisterMethod4EagerPhysicalBlob()

    blob_trait.RegisterBlobOperatorTraitMethod(
        oneflow._oneflow_internal.EagerPhysicalBlob)
    blob_trait.RegisterBlobOperatorTraitMethod(
        oneflow._oneflow_internal.ConsistentBlob)
    blob_trait.RegisterBlobOperatorTraitMethod(
        oneflow._oneflow_internal.MirroredBlob)

    remote_blob_util.RegisterMethod4EagerBlobTrait()
    remote_blob_util.RegisterMethod4LazyConsistentBlob()
    remote_blob_util.RegisterMethod4LazyMirroredBlob()
    remote_blob_util.RegisterMethod4EagerConsistentBlob()