Searched refs:multi_output_reduce_window (Results 1 – 1 of 1) sorted by relevance
5605 const bool multi_output_reduce_window = reduce_window->shape().IsTuple(); in HandleReduceWindow() local5612 CHECK(multi_output_reduce_window || elements.size() == 1); in HandleReduceWindow()5613 if (multi_output_reduce_window) { in HandleReduceWindow()5633 (!multi_output_reduce_window || in HandleReduceWindow()5644 if (multi_output_reduce_window) { in HandleReduceWindow()5657 if (multi_output_reduce_window && input_count == 1 && in HandleReduceWindow()5678 if (multi_output_reduce_window) { in HandleReduceWindow()