Function permute

src/framework/modules/whisper_embedding.cpp:17–22

}

core::TensorValue permute(
    core::ModuleBuildContext & ctx,
    const core::TensorValue & input,
    std::array<int, core::kMaxTensorRank> axes) {
    return TransposeModule({axes, input.shape.rank}).build(ctx, input);
}

core::TensorValue reshape_heads(
    core::ModuleBuildContext & ctx,