diff --git a/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py b/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py
index e4b8397016..7bb4923bb5 100755
--- a/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py
+++ b/mindspore/_extends/parallel_compile/tbe_compiler/compiler.py
@@ -107,7 +107,7 @@ def build_op(build_type, json_str):
             raise ValueError("Op:{} function {} is not supported by Tbe.".format(op_name, build_type))
 
         # call function
-        if kernel_name[0:19] == "bounding_box_encode":
+        if op_name == "bounding_box_encode":
             return op_func(*inputs_args, *outputs_args, *attrs_args, kernel_name_val=kernel_name)
 
         if is_dynamic_shape: