Searched defs:reduce_window (Results 1 – 16 of 16) sorted by relevance
97 HloInstruction *reduce_window = in HandleReduce() local
142 auto* reduce_window = in ExpandInstruction() local
402 auto reduce_window = add(HloInstruction::CreateReduceWindow( in HandleBatchGroupCount() local
199 Status HandleReduceWindow(HloInstructionPtr reduce_window) override { in HandleReduceWindow()
1419 auto* reduce_window = Cast<HloReduceWindowInstruction>(&user); in GetShardingFromUser() local1997 auto* reduce_window = Cast<HloReduceWindowInstruction>(instruction); in InferShardingFromOperands() local
410 const HloInstruction* reduce_window) { in HandleReduceWindow()
978 auto* reduce_window = in TEST_F() local
1526 auto* reduce_window = FindInstruction(module.get(), "reduce-window"); in TEST_P() local1579 auto* reduce_window = FindInstruction(module.get(), "reduce-window"); in TEST_P() local
104 auto* reduce_window = Cast<HloReduceWindowInstruction>(inst); in ChooseIdentityValue() local
1236 auto* reduce_window = Cast<HloReduceWindowInstruction>(hlo); in HandleReduceWindow() local
1262 Status ShapeVerifier::HandleReduceWindow(HloInstruction* reduce_window) { in HandleReduceWindow()
2784 const HloReduceWindowInstruction* reduce_window, in EmitElementalReduceWindow()
5604 auto* reduce_window = Cast<HloReduceWindowInstruction>(hlo); in HandleReduceWindow() local
416 def reduce_window(operand, function
666 Status IrEmitter::HandleReduceWindow(HloInstruction* reduce_window) { in HandleReduceWindow()
3898 HloReduceWindowInstruction* reduce_window = in HandleReduceWindow() local