fix dtype type error (#3289)

Former-commit-id: 2e33369
Oneflow-Inc · Jul 25, 2020 · b93f0d7 · b93f0d7
1 parent 93f8e2a
commit b93f0d7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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