Searched refs:reduce_window_instr (Results 1 – 3 of 3) sorted by relevance
1968 auto* reduce_window_instr = Cast<HloReduceWindowInstruction>(reduce_window);1974 reduce_window_instr->input_shapes(),1975 reduce_window_instr->init_value_shapes(), window,1985 auto input_arrays = reduce_window_instr->inputs();1986 auto init_values = reduce_window_instr->init_values();
1231 auto reduce_window_instr = Cast<HloReduceWindowInstruction>(reduce_window); in HandleReduceWindow() local1232 auto input_shapes = reduce_window_instr->input_shapes(); in HandleReduceWindow()1234 auto init_shapes = reduce_window_instr->init_value_shapes(); in HandleReduceWindow()
2679 auto reduce_window_instr = Cast<HloReduceWindowInstruction>(hlo); in MakeElementGenerator() local2681 for (const HloInstruction* instr : reduce_window_instr->inputs()) { in MakeElementGenerator()2686 for (const HloInstruction* instr : reduce_window_instr->init_values()) { in MakeElementGenerator()