onnxruntime

1
/**
2
 * @license
3
 * Copyright 2021 Google LLC. All Rights Reserved.
4
 * Licensed under the Apache License, Version 2.0 (the "License");
5
 * you may not use this file except in compliance with the License.
6
 * You may obtain a copy of the License at
7
 *
8
 * http://www.apache.org/licenses/LICENSE-2.0
9
 *
10
 * Unless required by applicable law or agreed to in writing, software
11
 * distributed under the License is distributed on an "AS IS" BASIS,
12
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
 * See the License for the specific language governing permissions and
14
 * limitations under the License.
15
 * =============================================================================
16
 */
17

18
// sampled from [@tensorflow/tfjs] tfjs-backend-webgpu/src/activation_util.ts
19
//
20
// modified to fit the needs of the project
21

22
export const typeSnippet = (component: number, dataType: string) => {
23
  switch (component) {
24
    case 1:
25
      return dataType;
26
    case 2:
27
      return `vec2<${dataType}>`;
28
    case 3:
29
      return `vec3<${dataType}>`;
30
    case 4:
31
      return `vec4<${dataType}>`;
32
    default:
33
      throw new Error(`${component}-component is not supported.`);
34
  }
35
};
36

37
export const biasSnippet = (hasBias: boolean): string => `
38
      ${hasBias ? 'value = value + getBiasByOutputCoords(coords);' : ''}
39
      `;
40