experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix: modify comment

Browse Source
This commit is contained in:
Wang, Yanzhang 2020-12-11 11:37:04 +08:00
parent 4588463244
commit d4ddd88395
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/grappler/optimizers/remapper.cc
View File

@ -1290,8 +1290,8 @@ Status AddFusedContractionNode(RemapperContext* ctx,
const NodeDef& add = graph->node(matched.add);
contraction_node.set_name(add.name());
contraction_node.set_device(contraction.device());
contraction_node.add_input(contraction.input(0)); // 0: input
contraction_node.add_input(contraction.input(1)); // 1: filter
contraction_node.add_input(contraction.input(0)); // 0: input(conv) / a (matmul)
contraction_node.add_input(contraction.input(1)); // 1: filter(conv) / b (matmul)
contraction_node.add_input(bias_add.input(1)); // 2: bias
// Add OP has two inputs, one is conv+bias/matmul+bias pattern matched