Searched refs:bias_broadcast_instr (Results 1 – 1 of 1) sorted by relevance
75 HloInstruction* bias_broadcast_instr = nullptr; in FindConvWithRelu() local84 const auto bias_pattern = Broadcast(&bias_broadcast_instr, Op(&bias)); in FindConvWithRelu()143 {conv_instr, bias_broadcast_instr, gte, add1, add2, mul1, mul2}) { in FindConvWithRelu()151 CastOrNull<HloBroadcastInstruction>(bias_broadcast_instr); in FindConvWithRelu()167 if (bias_broadcast_instr->dimensions().size() != 1) { in FindConvWithRelu()170 if (bias_broadcast_instr->dimensions(0) != in FindConvWithRelu()