def _CreateEagerVariableBlob(op_attribute):
bn_in_op2blob_object = {}
def BuildInstruction(builder):
parallel_conf = oneflow.placement.current_scope().default_parallel_conf
builder.StatelessCall(op_attribute,
parallel_conf,
bn_in_op2blob_object=bn_in_op2blob_object)
vm_util.LogicalRun(BuildInstruction)
lbi = logical_blob_id_util.LogicalBlobId()
lbi.op_name = op_attribute.op_conf.name
lbi.blob_name = op_attribute.op_conf.variable_conf.out
return remote_blob_util.EagerLogicalBlob(
lbi, blob_object=bn_in_op2blob_object["out"])
def MakeRemoteBlob(self, lbi):
return remote_blob_util.EagerLogicalBlob(lbi)
