Skip to content

chore(api): refresh functiongemma/gemma dumps for #415/#417's new API surface #422

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
michalharakal wants to merge 1 commit into from
Closed

chore(api): refresh functiongemma/gemma dumps for #415/#417's new API surface #422

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 19 additions & 0 deletions llm-inference/functiongemma/api/functiongemma.api
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,14 +1,23 @@
public final class sk/ainet/models/functiongemma/FunctionGemmaContract {
public static final field CONTRACT_VERSION I
public static final field DEFAULT_CHUNK I
public static final field FN_PREFILL Ljava/lang/String;
public static final field FN_PREFILL_AT Ljava/lang/String;
public static final field FN_PREFILL_WITH_PAST Ljava/lang/String;
public static final field FN_REDECODE Ljava/lang/String;
public static final field FN_REDECODE_AT Ljava/lang/String;
public static final field FN_WITH_PAST Ljava/lang/String;
public static final field INSTANCE Lsk/ainet/models/functiongemma/FunctionGemmaContract;
public static final field K_FIRST_IN_OUTPUT Z
public static final field PARAMETER_SCOPE Ljava/lang/String;
public final fun manifestJson (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/lang/String;
public final fun prefillArgs ()Ljava/util/List;
public final fun prefillAtOutputs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List;
public final fun prefillOutputs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List;
public final fun prefillWithPastArgs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List;
public final fun prefillWithPastOutputs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List;
public final fun qualified (Ljava/lang/String;)Ljava/lang/String;
public final fun selectArgs ()Ljava/util/List;
public final fun withPastArgs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List;
public final fun withPastOutputs (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;)Ljava/util/List;
}
Expand All @@ -25,7 +34,17 @@ public final class sk/ainet/models/functiongemma/FunctionGemmaExportHarness {
public final fun exportPrefill (Ljava/lang/String;Ljava/lang/String;IFZ)Ljava/lang/String;
public final fun exportPrefill (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;)Ljava/lang/String;
public static synthetic fun exportPrefill$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Ljava/lang/String;Ljava/lang/String;IFZILjava/lang/Object;)Ljava/lang/String;
public final fun exportPrefillAt (Ljava/lang/String;Ljava/lang/String;IFZ)Ljava/lang/String;
public final fun exportPrefillAt (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;)Ljava/lang/String;
public static synthetic fun exportPrefillAt$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Ljava/lang/String;Ljava/lang/String;IFZILjava/lang/Object;)Ljava/lang/String;
public final fun exportPrefillWithPast (Ljava/lang/String;Ljava/lang/String;IFZ)Ljava/lang/String;
public final fun exportPrefillWithPast (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;I)Ljava/lang/String;
public static synthetic fun exportPrefillWithPast$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Ljava/lang/String;Ljava/lang/String;IFZILjava/lang/Object;)Ljava/lang/String;
public static synthetic fun exportPrefillWithPast$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;IILjava/lang/Object;)Ljava/lang/String;
public final fun exportRedecode (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;)Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness$RedecodeResult;
public final fun exportRedecodeAt (Ljava/lang/String;Ljava/lang/String;IFZ)Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness$RedecodeResult;
public final fun exportRedecodeAt (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;)Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness$RedecodeResult;
public static synthetic fun exportRedecodeAt$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Ljava/lang/String;Ljava/lang/String;IFZILjava/lang/Object;)Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness$RedecodeResult;
public final fun exportWithPast (Ljava/lang/String;Ljava/lang/String;IZFZ)Ljava/lang/String;
public final fun exportWithPast (Lsk/ainet/models/functiongemma/FunctionGemmaSpec;Ljava/lang/String;)Ljava/lang/String;
public static synthetic fun exportWithPast$default (Lsk/ainet/models/functiongemma/FunctionGemmaExportHarness;Ljava/lang/String;Ljava/lang/String;IZFZILjava/lang/Object;)Ljava/lang/String;
Expand Down
13 changes: 13 additions & 0 deletions llm-inference/gemma/api/jvm/gemma.api
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,10 @@ public final class sk/ainet/models/gemma/GemmaModel : sk/ainet/lang/nn/Module {
public fun <init> (Lsk/ainet/lang/nn/layers/EmbeddingAdapter;Lsk/ainet/models/gemma/PerLayerEmbedding;Ljava/util/List;Lsk/ainet/lang/nn/normalization/RMSNormalization;Lsk/ainet/lang/nn/transformer/VoidDense;Lkotlin/reflect/KClass;FFLjava/lang/String;)V
public synthetic fun <init> (Lsk/ainet/lang/nn/layers/EmbeddingAdapter;Lsk/ainet/models/gemma/PerLayerEmbedding;Ljava/util/List;Lsk/ainet/lang/nn/normalization/RMSNormalization;Lsk/ainet/lang/nn/transformer/VoidDense;Lkotlin/reflect/KClass;FFLjava/lang/String;ILkotlin/jvm/internal/DefaultConstructorMarker;)V
public final fun buildRopeCosSin (ILsk/ainet/context/ExecutionContext;)Lsk/ainet/models/gemma/GemmaModel$RopeCosSin;
public final fun forwardAt (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/context/ExecutionContext;)Lsk/ainet/lang/tensor/Tensor;
public final fun forwardPrefill (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/context/ExecutionContext;)Lsk/ainet/models/gemma/GemmaModel$GemmaPrefillOutput;
public final fun forwardPrefillAt (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/context/ExecutionContext;)Lsk/ainet/models/gemma/GemmaModel$GemmaPrefillOutput;
public final fun forwardPrefillWithPast (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/models/gemma/GemmaModel$ChunkContext;Lsk/ainet/lang/tensor/Tensor;Ljava/util/List;Ljava/util/List;Lsk/ainet/context/ExecutionContext;)Lsk/ainet/models/gemma/GemmaModel$GemmaWithPastOutput;
public final fun forwardWithPast (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/models/gemma/GemmaModel$RopeCosSin;Ljava/util/List;Ljava/util/List;Lsk/ainet/context/ExecutionContext;)Lsk/ainet/models/gemma/GemmaModel$GemmaWithPastOutput;
public final fun getBlocks ()Ljava/util/List;
public final fun getDtype ()Lkotlin/reflect/KClass;
Expand All @@ -63,6 +66,16 @@ public final class sk/ainet/models/gemma/GemmaModel : sk/ainet/lang/nn/Module {
public final fun getTokenEmbedding ()Lsk/ainet/lang/nn/layers/EmbeddingAdapter;
}

public final class sk/ainet/models/gemma/GemmaModel$ChunkContext {
public fun <init> (Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;Lsk/ainet/lang/tensor/Tensor;)V
public final fun getCosGlobal ()Lsk/ainet/lang/tensor/Tensor;
public final fun getCosSliding ()Lsk/ainet/lang/tensor/Tensor;
public final fun getMaskGlobal ()Lsk/ainet/lang/tensor/Tensor;
public final fun getMaskSliding ()Lsk/ainet/lang/tensor/Tensor;
public final fun getSinGlobal ()Lsk/ainet/lang/tensor/Tensor;
public final fun getSinSliding ()Lsk/ainet/lang/tensor/Tensor;
}

public final class sk/ainet/models/gemma/GemmaModel$Companion {
public final fun findHookIn (Ljava/util/List;)Lsk/ainet/models/gemma/PerLayerInputBlockHook;
}
Expand Down