Searched refs:input_hlos (Results 1 – 1 of 1) sorted by relevance
2746 std::vector<HloInstruction*> input_hlos(input_count); in HandleReduce() local2748 input_hlos[i] = inputs[i].hlo(); in HandleReduce()2751 reduce_shape, input_hlos, inits, hlo->dimensions(), hlo->to_apply())); in HandleReduce()2779 auto expanded_shape = input_hlos[i]->shape(); in HandleReduce()2780 auto all_gather_shape = input_hlos[i]->shape(); in HandleReduce()