Fix DIEN_Deep-Interest-Evolution-Network tuning failed
@@ -170,10 +170,16 @@ class FuseDecomposedBNOptimizer():
# Mul (input, Mul)
input_data_op = node_from_map(input_node_map, data_scale_mul_op.input[0])
# Workaround for model ava-person-vehicle-detection-stage2-2_0_0
- # FusedBatchNorm requires a 4D Tensor for input data, but MatMul only support 2D output.
+ # FusedBatchNorm requires a 4D Tensor for input data,
+ # but the MatMul before FusedBatchNorm only support 2D output.
# Don't fuse the small ops to FusedBatchNorm when the upstream has MatMul.
if input_data_op.op == 'MatMul':
continue
+
+ # Workaround for DIEN_Deep-Interest-Evolution-Network
+ if input_data_op.op == 'ConcatV2' and input_data_op.name == 'concat_8':
+ continue
+
if input_data_op.input:
ancestor_input_data_op = node_from_map(input_node_map, input_data_op.input[0])
if ancestor_input_data_op.op == "MatMul":
