Searched refs:tuple_of_instructions (Results 1 – 2 of 2) sorted by relevance
1584 const Shape& shape, HloInstruction* tuple_of_instructions, in CreateReduce() argument1588 if (!tuple_of_instructions->shape().IsTuple()) { in CreateReduce()1592 return CreateReduce(shape, tuple_of_instructions, init_values[0], in CreateReduce()1596 for (int idx = 0; idx < tuple_of_instructions->shape().tuple_shapes_size(); in CreateReduce()1599 HloInstruction::CreateGetTupleElement(tuple_of_instructions, idx); in CreateReduce()1601 tuple_of_instructions->parent()->AddInstruction(std::move(gte))); in CreateReduce()
938 const Shape& shape, HloInstruction* tuple_of_instructions,