blob: 7a6cab72b4201164ae3571c69e9ea41815b4a764 (
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
|
# Simple Shape network
node {
name: "Const"
op: "Const"
attr {
key: "dtype"
value {
type: DT_FLOAT
}
}
attr {
key: "value"
value {
tensor {
dtype: DT_FLOAT
tensor_shape {
dim {
size: 2
}
dim {
size: 3
}
}
float_val: 1.0
}
}
}
}
node {
name: "Shape"
op: "Shape"
input: "Const"
attr {
key: "T"
value { type: DT_FLOAT }
}
attr {
key: "out_type"
value { type: DT_INT32 }
}
}
|