File tree Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -130,8 +130,8 @@ class ConvMKLDNNHandler : public platform::MKLDNNHandler {
130
130
131
131
std::shared_ptr<mkldnn::memory> AcquireWeightsMemoryFromPrimitive (
132
132
const std::shared_ptr<mkldnn::memory> user_weights_memory_p,
133
- const std::vector<mkldnn::primitive>& pipeline,
134
- bool is_persistent = false ) { // NOLINT
133
+ std::vector<mkldnn::primitive>& pipeline, // NOLINT
134
+ bool is_persistent = false ) {
135
135
auto user_weights_pd = user_weights_memory_p->get_primitive_desc ();
136
136
auto weights_pd = conv_pd_->weights_primitive_desc ();
137
137
return this ->AcquireMemory (weights_pd, user_weights_pd,
Original file line number Diff line number Diff line change @@ -191,8 +191,9 @@ class MKLDNNHandler {
191
191
mkldnn::memory::primitive_desc& mpd, // NOLINT
192
192
mkldnn::memory::primitive_desc& user_mpd, // NOLINT
193
193
const std::shared_ptr<mkldnn::memory> user_memory_p,
194
- const std::string& suffix, const std::vector<mkldnn::primitive>& pipeline,
195
- bool is_persistent = false ) { // NOLINT
194
+ const std::string& suffix,
195
+ std::vector<mkldnn::primitive>& pipeline, // NOLINT
196
+ bool is_persistent = false ) {
196
197
// create reorder primitive if the input format is not the preferred one
197
198
auto local_key = key_ + suffix;
198
199
auto key_reorder_p = key_ + suffix + " reorder_p" ;
You can’t perform that action at this time.
0 commit comments