Skip to content

N
ngraph
Commit 716fd2e9 authored by Pruthvi Gowda Thorehosur Appajigowda's avatar Pruthvi Gowda Thorehosur Appajigowda
Browse files
Options

Added MaxPoolBackprop to the mkldnn_util

parent 3208c02b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/ngraph/runtime/cpu/mkldnn_utils.cpp
View file @ 716fd2e9
...@@ -41,7 +41,8 @@ namespace ngraph ...@@ -41,7 +41,8 @@ namespace ngraph
TI(ngraph::op::Convolution), TI(ngraph::op::Convolution),
TI(ngraph::op::ConvolutionBackpropData), TI(ngraph::op::ConvolutionBackpropData),
TI(ngraph::op::ConvolutionBackpropFilters), TI(ngraph::op::ConvolutionBackpropFilters),
TI(ngraph::op::MaxPool)}; TI(ngraph::op::MaxPool),
T1(ngraph::op::MaxPoolBackprop)};
bool IsMKLDNNOp(ngraph::Node& op) bool IsMKLDNNOp(ngraph::Node& op)
{ {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment