mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
Summary: https://github.com/pytorch/pytorch/pull/51613 made `torch.nn.functional.linear` compile as `aten::linear`, extend the same behavior with tracing. Pull Request resolved: https://github.com/pytorch/pytorch/pull/51897 Reviewed By: albanD Differential Revision: D26320711 Pulled By: eellison fbshipit-source-id: a26d3c37323a0706313c6ebb210bad60eec6a64b
104 lines
No EOL
1.4 KiB
Text
104 lines
No EOL
1.4 KiB
Text
node {
|
|
name: "input/x"
|
|
op: "IO Node"
|
|
attr {
|
|
key: "_output_shapes"
|
|
value {
|
|
list {
|
|
shape {
|
|
dim {
|
|
size: 1
|
|
}
|
|
dim {
|
|
size: 3
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
attr {
|
|
key: "attr"
|
|
value {
|
|
s: ""
|
|
}
|
|
}
|
|
}
|
|
node {
|
|
name: "output/output.1"
|
|
op: "IO Node"
|
|
input: "myLinear/Linear[l]/18"
|
|
attr {
|
|
key: "_output_shapes"
|
|
value {
|
|
list {
|
|
shape {
|
|
dim {
|
|
size: 1
|
|
}
|
|
dim {
|
|
size: 5
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
attr {
|
|
key: "attr"
|
|
value {
|
|
s: ""
|
|
}
|
|
}
|
|
}
|
|
node {
|
|
name: "myLinear/Linear[l]/bias/16"
|
|
op: "prim::GetAttr"
|
|
input: "myLinear/Linear[l]/weight/13"
|
|
attr {
|
|
key: "attr"
|
|
value {
|
|
s: "{ name : bias }"
|
|
}
|
|
}
|
|
}
|
|
node {
|
|
name: "myLinear/Linear[l]/weight/17"
|
|
op: "prim::GetAttr"
|
|
input: "myLinear/Linear[l]/weight/13"
|
|
attr {
|
|
key: "attr"
|
|
value {
|
|
s: "{ name : weight }"
|
|
}
|
|
}
|
|
}
|
|
node {
|
|
name: "myLinear/Linear[l]/18"
|
|
op: "aten::linear"
|
|
input: "input/x"
|
|
input: "myLinear/Linear[l]/weight/17"
|
|
input: "myLinear/Linear[l]/bias/16"
|
|
attr {
|
|
key: "_output_shapes"
|
|
value {
|
|
list {
|
|
shape {
|
|
dim {
|
|
size: 1
|
|
}
|
|
dim {
|
|
size: 5
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
attr {
|
|
key: "attr"
|
|
value {
|
|
s: "{}"
|
|
}
|
|
}
|
|
}
|
|
versions {
|
|
producer: 22
|
|
} |