|
|
@ -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))
|
|
|
|
raise ValueError("Op:{} function {} is not supported by Tbe.".format(op_name, build_type))
|
|
|
|
|
|
|
|
|
|
|
|
# call function
|
|
|
|
# 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)
|
|
|
|
return op_func(*inputs_args, *outputs_args, *attrs_args, kernel_name_val=kernel_name)
|
|
|
|
|
|
|
|
|
|
|
|
if is_dynamic_shape:
|
|
|
|
if is_dynamic_shape:
|
|
|
|