Skip to content

Commit 056e3d8

Browse files
committed
fix pylint error
1 parent 4c8661d commit 056e3d8

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tf2onnx/function/sparse_softmax_cross_entropy_with_logits.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ def softmax_cross_entropy_with_logits_computation(ctx, label, logit, tf_ori_node
2828
shapes = tf_ori_node.output_shapes
2929
dtypes = tf_ori_node.output_dtypes
3030
ctx.remove_node(tf_ori_node.name)
31-
res = ctx.make_node(op_type="Squeeze", inputs=[mul2.output[0]], attr={"axes": [1]},
32-
outputs=[tf_ori_node.output[0]], shapes=[shapes[0]], dtypes=[dtypes[0]])
31+
ctx.make_node(op_type="Squeeze", inputs=[mul2.output[0]], attr={"axes": [1]},
32+
outputs=[tf_ori_node.output[0]], shapes=[shapes[0]], dtypes=[dtypes[0]])
3333

3434

3535
def softmax_cross_entropy_with_logits_op(ctx, node, name, args):
@@ -151,4 +151,3 @@ def sparse_softmax_cross_entropy_with_logits_op9(ctx, node, name, args):
151151
label_node = ctx.make_node("Cast", label_node.output, attr={"to": logit_dtype}, dtypes=[logit_dtype])
152152

153153
softmax_cross_entropy_with_logits_computation(ctx, label_node, logit_node, node)
154-

0 commit comments

Comments
 (0)