Searched refs:input_array_shapes (Results 1 – 3 of 3) sorted by relevance
895 auto input_shapes = reduce_window_instr->input_array_shapes(); in HandleReduceWindow()
1318 absl::InlinedVector<const Shape*, 2> input_array_shapes() const { in input_array_shapes() function
1952 reduce_window_instr->input_array_shapes(),