def BuildInstruction(builder): with blob_register_util.BnInOp2BlobObjectScope( blob_register, op_attribute) as bn_in_op2blob_object: physical_out_blob_objects = builder.UnpackLogicalBlobToPhysicalBlobs( GetInBlobObject(builder, "in", bn_in_op2blob_object)) for i, blob_object in enumerate(physical_out_blob_objects): bn_in_op2blob_object["out_%s" % i] = blob_object
def BuildInstruction(builder): with blob_register_util.BnInOp2BlobObjectScope( blob_register, op_attribute) as bn_in_op2blob_object: builder.StatefulCall( op_attribute, opkernel_object=opkernel_object, bn_in_op2blob_object=bn_in_op2blob_object, )
def BuildInstruction(builder): with blob_register_util.BnInOp2BlobObjectScope( blob_register, op_attribute) as bn_in_op2blob_object: builder.StatelessCall( op_attribute, parallel_conf, bn_in_op2blob_object=bn_in_op2blob_object, )
def BuildInstruction(builder): with blob_register_util.BnInOp2BlobObjectScope( blob_register, op_attribute) as bn_in_op2blob_object: in_blob_object = bn_in_op2blob_object["in"] parallel_desc_symbol = in_blob_object.parallel_desc_symbol op_arg_parallel_attr = oneflow_api.GetOpArgParallelAttribute( parallel_desc_symbol, str(op_attribute), "out") out_blob_object = builder.MakeReferenceBlobObject( in_blob_object, op_arg_parallel_attr) bn_in_op2blob_object["out"] = out_blob_object
def BuildInstruction(builder): with blob_register_util.BnInOp2BlobObjectScope( blob_register, op_attribute) as bn_in_op2blob_object: cfg_op_attribute = oneflow_api.deprecated.MakeOpAttributeByString( str(op_attribute)) builder.StatefulCall( cfg_op_attribute, opkernel_object, bn_in_op2blob_object, boxing_util.BoxingTo, )
def BuildInstruction(builder): with blob_register_util.BnInOp2BlobObjectScope( blob_register, op_attribute) as bn_in_op2blob_object: cfg_op_attribute = oneflow._oneflow_internal.deprecated.MakeOpAttributeByString( str(op_attribute)) builder.StatelessCall( cfg_op_attribute, parallel_conf, bn_in_op2blob_object, boxing_util.BoxingTo, )
def BuildInstruction(builder): with blob_register_util.BnInOp2BlobObjectScope( blob_register, op_attribute ) as bn_in_op2blob_object: def GetPhysicalInBlob(i): return GetInBlobObject(builder, i, bn_in_op2blob_object) in_blob_objects = [GetPhysicalInBlob(i) for i in range(parallel_size)] bn_in_op2blob_object["out"] = builder.PackPhysicalBlobsToLogicalBlob( in_blob_objects, op_arg_parallel_attr, op_arg_blob_attr )