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)