blob: b2cd68f9fdbc9f74416108e9c11251bd61e4c053 (
plain)
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
|
operand {
name: "ifm"
type: FLOAT32
shape { dim: 3 dim: 8 dim: 8 dim: 4 }
}
operand {
name: "inner"
type: FLOAT32
shape { dim: 3 dim: 4 }
}
operand {
name: "reduction_indices1"
type: INT32
shape { dim: 2 }
filler { tag: "explicit" arg: "1" arg: "2" }
}
operand {
name: "reduction_indices2"
type: INT32
shape { dim: 1 }
filler { tag: "explicit" arg: "1"}
}
operand {
name: "ofm"
type: FLOAT32
shape { dim: 3 }
}
operation {
type: "Mean"
mean_options {
keep_dims: false
}
input: "ifm"
input: "reduction_indices1"
output: "inner"
}
operation {
type: "Mean"
mean_options {
keep_dims: false
}
input: "inner"
input: "reduction_indices2"
output: "ofm"
}
input: "ifm"
output: "ofm"
|