提交 04e4f6d9 编写于 作者: Z zhaozhenlong 提交者: 高东海

adapt relu6grad and graphengine modified

上级 02ebf03c
graphengine @ 78a40081
Subproject commit 092c7a1f6548cac7d40e677af3498c3c49ea2bfd
Subproject commit 78a40081fc1462ab9e2083d37e5fbe608a57dd06
......@@ -473,7 +473,7 @@ ATTR_MAP(Relu6) = EMPTY_ATTR_MAP;
OUTPUT_MAP(Relu6) = {{0, OUTPUT_DESC(activations)}};
// Relu6Grad
INPUT_MAP(Relu6Grad) = {{1, INPUT_DESC(features)}, {2, INPUT_DESC(gradients)}};
INPUT_MAP(Relu6Grad) = {{1, INPUT_DESC(gradients)}, {2, INPUT_DESC(features)}};
ATTR_MAP(Relu6Grad) = EMPTY_ATTR_MAP;
OUTPUT_MAP(Relu6Grad) = {{0, OUTPUT_DESC(backprops)}};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册