Searched refs:relu_input (Results 1 – 2 of 2) sorted by relevance
66 Operation *relu_input = relu_op.features().getDefiningOp(); in matchAndRewrite() local67 if (!relu_input) return failure(); in matchAndRewrite()68 auto batch_norm = dyn_cast_or_null<FusedBatchNormV3Op>(relu_input); in matchAndRewrite()74 add_op = dyn_cast_or_null<AddV2Op>(relu_input); in matchAndRewrite()
64 HloInstruction* relu_input; in FindConvWithRelu() local68 if (!Match(instr, Maximum(zero_pattern, Op(&relu_input))) && in FindConvWithRelu()69 !Match(instr, Maximum(Op(&relu_input), zero_pattern))) { in FindConvWithRelu()122 CHECK(Match(relu_input, add3_pattern)); in FindConvWithRelu()