blob: 1c760c87f0a429ea314113e4c2afe660ae996279 (
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
|
# Network with Squeeze that have multiple negative squeeze dimensions
node {
name: "Placeholder"
op: "Placeholder"
attr {
key: "dtype"
value { type: DT_FLOAT }
}
attr {
key: "shape"
value {
shape {
dim { size: 2 }
dim { size: 1 }
dim { size: 3 }
dim { size: 1 }
}
}
}
}
node {
name: "Squeeze"
op: "Squeeze"
input: "Placeholder"
attr {
key: "T"
value { type: DT_FLOAT }
}
attr {
key: "squeeze_dims"
value {
list { i: -1 i: -3 }
}
}
}
|