Searched refs:multi_output_reduce (Results 1 – 1 of 1) sorted by relevance
2910 bool multi_output_reduce = reduce->shape().IsTuple(); in HandleReduce() local2918 multi_output_reduce ? reduce->shape().tuple_shapes(0) : reduce->shape(); in HandleReduce()2924 if (multi_output_reduce) { in HandleReduce()2947 if (multi_output_reduce) { in HandleReduce()2965 if (multi_output_reduce) { in HandleReduce()