Searched refs:reduce_window_instr (Results 1 – 3 of 3) sorted by relevance
894 auto reduce_window_instr = Cast<HloReduceWindowInstruction>(reduce_window); in HandleReduceWindow() local895 auto input_shapes = reduce_window_instr->input_array_shapes(); in HandleReduceWindow()897 auto init_shapes = reduce_window_instr->init_value_shapes(); in HandleReduceWindow()
1946 auto* reduce_window_instr = Cast<HloReduceWindowInstruction>(reduce_window);1952 reduce_window_instr->input_array_shapes(),1953 reduce_window_instr->init_value_shapes(), window,1963 auto input_arrays = reduce_window_instr->input_arrays();1964 auto init_values = reduce_window_instr->init_values();
2493 auto reduce_window_instr = Cast<HloReduceWindowInstruction>(hlo); in MakeElementGenerator() local2496 reduce_window_instr->input_arrays()) { in MakeElementGenerator()2501 for (const HloInstruction* instr : reduce_window_instr->init_values()) { in MakeElementGenerator()