Searched refs:mask_amount (Results 1 – 1 of 1) sorted by relevance
2490 auto* mask_amount = in TryRemainderToAnd() local2494 mask_amount = computation->AddInstruction( in TryRemainderToAnd()2495 HloInstruction::CreateBroadcast(b->shape(), mask_amount, {})); in TryRemainderToAnd()2499 remainder->shape(), HloOpcode::kAnd, abs_dividend, mask_amount)); in TryRemainderToAnd()2512 HloInstruction* mask_amount = in TryRemainderToAnd() local2516 mask_amount = computation->AddInstruction( in TryRemainderToAnd()2517 HloInstruction::CreateBroadcast(b->shape(), mask_amount, {})); in TryRemainderToAnd()2520 a, mask_amount); in TryRemainderToAnd()