Searched defs:reduce_window (Results 1 – 11 of 11) sorted by relevance
167 Status HandleReduceWindow(HloInstructionPtr reduce_window) override { in HandleReduceWindow()
300 auto reduce_window = add(HloInstruction::CreateReduceWindow( in HandleBatchGroupCount() local
585 auto* reduce_window = in TEST_F() local
356 HloInstruction* reduce_window = hlo; in HandleReduceWindow() local
303 const HloInstruction* reduce_window) { in HandleReduceWindow()
622 Status ShapeVerifier::HandleReduceWindow(HloInstruction* reduce_window) { in HandleReduceWindow()
1961 Status HandleReduceWindow(HloInstruction* reduce_window) override { in HandleReduceWindow()
3084 HloInstruction* reduce_window) { in HandleReduceWindow()
3619 HloInstruction* reduce_window = in TEST_F() local3705 HloInstruction* reduce_window = in TEST_F() local
350 def reduce_window(operand, function
641 const HloReduceWindowInstruction* reduce_window, in EmitElementalReduceWindow()728 Status IrEmitter::HandleReduceWindow(HloInstruction* reduce_window) { in HandleReduceWindow()