Searched refs:tfl_add_op (Results 1 – 1 of 1) sorted by relevance
476 auto tfl_add_op = cast<TflOp>(op); in matchAndRewriteAddSub() local479 tfl_add_op.lhs().getType().template dyn_cast<ShapedType>(); in matchAndRewriteAddSub()481 tfl_add_op.rhs().getType().template dyn_cast<ShapedType>(); in matchAndRewriteAddSub()483 tfl_add_op.getResult().getType().template dyn_cast<ShapedType>(); in matchAndRewriteAddSub()540 buildRescaleToInt32(rewriter, op, tfl_add_op.lhs(), lhs_rescale_scale, in matchAndRewriteAddSub()543 buildRescaleToInt32(rewriter, op, tfl_add_op.rhs(), rhs_rescale_scale, in matchAndRewriteAddSub()554 tfl_add_op.lhs(), tfl_add_op.rhs()); in matchAndRewriteAddSub()559 auto fused_activation_fn = tfl_add_op.fused_activation_functionAttr(); in matchAndRewriteAddSub()