Searched refs:tuple_args (Results 1 – 2 of 2) sorted by relevance
1054 auto tuple_args = std::forward_as_tuple(std::forward<Args>(args)...);1056 a.Perform(tuple_args);1058 return last.Perform(std::move(tuple_args));
2772 std::vector<HloInstruction*> tuple_args(tuple_shape.tuple_shapes_size()); in ReplaceTupleWith() local2790 TF_ASSIGN_OR_RETURN(tuple_args[i], in ReplaceTupleWith()2811 tuple_args[i] = new_instruction; in ReplaceTupleWith()2814 tuple_args[i] = get_operand(); in ReplaceTupleWith()2817 return computation->AddInstruction(HloInstruction::CreateTuple(tuple_args)); in ReplaceTupleWith()