wuhp commited on
Commit
a380586
·
verified ·
1 Parent(s): 05e0780

Update types.ts

Browse files
Files changed (1) hide show
  1. types.ts +4 -1
types.ts CHANGED
@@ -31,6 +31,9 @@ export enum LayerType {
31
  RMSNORM = 'RMSNorm',
32
  INSTANCENORM = 'InstanceNorm2d',
33
  FLATTEN = 'Flatten',
 
 
 
34
  EMBEDDING = 'Embedding',
35
  PATCH_EMBED = 'PatchEmbed',
36
  POS_EMBED = 'PositionalEmbed',
@@ -184,4 +187,4 @@ export interface LogEntry {
184
  timestamp: Date;
185
  message: string;
186
  type: 'info' | 'success' | 'warning' | 'error';
187
- }
 
31
  RMSNORM = 'RMSNorm',
32
  INSTANCENORM = 'InstanceNorm2d',
33
  FLATTEN = 'Flatten',
34
+ RESHAPE = 'Reshape',
35
+ PERMUTE = 'Permute',
36
+ UNFLATTEN = 'Unflatten',
37
  EMBEDDING = 'Embedding',
38
  PATCH_EMBED = 'PatchEmbed',
39
  POS_EMBED = 'PositionalEmbed',
 
187
  timestamp: Date;
188
  message: string;
189
  type: 'info' | 'success' | 'warning' | 'error';
190
+ }