diff --git a/oneflow/python/experimental/dynamic_binary_split_concat_op.py b/oneflow/python/experimental/dynamic_binary_split_concat_op.py index 2073c2f27af..57e68307bee 100644 --- a/oneflow/python/experimental/dynamic_binary_split_concat_op.py +++ b/oneflow/python/experimental/dynamic_binary_split_concat_op.py @@ -80,7 +80,9 @@ def dynamic_binary_concat( getattr(op_conf.dynamic_binary_concat_conf, "in").extend(in_lbns) # op_conf.dynamic_binary_concat_conf.in[:] = in_lbns op_conf.dynamic_binary_concat_conf.out = "out" - op_conf.dynamic_binary_concat_conf.out_data_type = source_blob.dtype + op_conf.dynamic_binary_concat_conf.out_data_type = ( + source_blob.dtype.oneflow_proto_dtype + ) op_conf.dynamic_binary_concat_conf.out_shape.dim.extend(list(source_blob.shape)) if source_blob.batch_axis is not None: op_conf.dynamic_binary_concat_conf.out_batch_axis.value = source_blob.batch_axis diff --git a/oneflow/python/ops/nn_ops.py b/oneflow/python/ops/nn_ops.py index caf152fb5e3..68e590ad678 100644 --- a/oneflow/python/ops/nn_ops.py +++ b/oneflow/python/ops/nn_ops.py @@ -670,7 +670,7 @@ def sigmoid_cross_entropy_with_logits( op_conf.sigmoid_cross_entropy_conf.prediction = logits.unique_name op_conf.sigmoid_cross_entropy_conf.label = labels.unique_name op_conf.sigmoid_cross_entropy_conf.loss = "loss" - op_conf.sigmoid_cross_entropy_conf.label_type = labels.dtype + op_conf.sigmoid_cross_entropy_conf.label_type = labels.dtype.oneflow_proto_dtype interpret_util.Forward(op_conf) lbi = logical_blob_id_util.LogicalBlobId() lbi.op_name = op_conf.name