gx

chenyc

2025-02-12 ea42ff3ebee1eeb3fb29423aa848a249441db81c

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import * as tf from '../../dist/tfjs.esm';
 
// eslint-disable-next-line no-unused-vars
export type ExtractWeightsFunction = (numWeights: number) => Float32Array
 
export type ParamMapping = {
  originalPath?: string
  paramPath: string
}
 
export type ConvParams = {
  filters: tf.Tensor4D
  bias: tf.Tensor1D
}
 
export type FCParams = {
  weights: tf.Tensor2D
  bias: tf.Tensor1D
}
 
export class SeparableConvParams {
  // eslint-disable-next-line no-useless-constructor
  constructor(
    // eslint-disable-next-line no-unused-vars
    public depthwise_filter: tf.Tensor4D,
    // eslint-disable-next-line no-unused-vars
    public pointwise_filter: tf.Tensor4D,
    // eslint-disable-next-line no-unused-vars
    public bias: tf.Tensor1D,
  // eslint-disable-next-line no-empty-function
  ) {}
}