ONE/res/TensorFlowLiteRecipes/Net_FullyConnected_Gelu_001/test.recipe at bc4ed0b3957ae5f418bcc1f1755a42a98ab413cc · mbencer/ONE · GitHub

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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
operand {
  name: "in"
  type: FLOAT32
  shape { dim: 1 dim: 16 }
}
operand {
  name: "weight"
  type: FLOAT32
  shape { dim: 16 dim: 16 }
  filler {
    tag: "gaussian"
    arg: "0.0"
    arg: "1.0"
  }
}
operand {
  name: "bias"
  type: FLOAT32
  shape { dim: 16 }
  filler {
    tag: "gaussian"
    arg: "0.0"
    arg: "1.0"
  }
}
operand {
  name: "out"
  type: FLOAT32
  shape { dim: 1 dim: 16 }
}
operation {
  type: "FullyConnected"
  fullyconnected_options {
    activation: RELU
  }
  input: "in"
  input: "weight"
  input: "bias"
  output: "out"
}
operand {
  name: "gelu_out"
  type: FLOAT32
  shape { dim: 1 dim: 16 }
}
operation {
  type: "Gelu"
  gelu_options {
    approximate: false
  }
  input: "out"
  output: "gelu_out"
}
operand {
  name: "weight_2"
  type: FLOAT32
  shape { dim: 4 dim: 16 }
  filler {
    tag: "gaussian"
    arg: "0.0"
    arg: "1.0"
  }
}
operand {
  name: "bias_2"
  type: FLOAT32
  shape { dim: 4 }
  filler {
    tag: "gaussian"
    arg: "0.0"
    arg: "1.0"
  }
}
operand {
  name: "out_2"
  type: FLOAT32
  shape { dim: 1 dim: 4 }
}
operation {
  type: "FullyConnected"
  fullyconnected_options {
    activation: RELU
  }
  input: "gelu_out"
  input: "weight_2"
  input: "bias_2"
  output: "out_2"
}
input: "in"
output: "out_2"