File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -131,12 +131,12 @@ class ConvMKLDNNHandler : public platform::MKLDNNHandler {
131
131
std::shared_ptr<mkldnn::memory> AcquireWeightsMemoryFromPrimitive (
132
132
const std::shared_ptr<mkldnn::memory> user_weights_memory_p,
133
133
const std::vector<mkldnn::primitive>& pipeline,
134
- bool is_test = false ) { // NOLINT
134
+ bool is_persistent = false ) { // NOLINT
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,
138
138
user_weights_memory_p, " @weights_mem_p" ,
139
- pipeline, is_test );
139
+ pipeline, is_persistent );
140
140
}
141
141
142
142
std::shared_ptr<mkldnn::memory> AcquireBiasMemoryFromPrimitive (
Original file line number Diff line number Diff line change @@ -192,7 +192,7 @@ class MKLDNNHandler {
192
192
mkldnn::memory::primitive_desc& user_mpd, // NOLINT
193
193
const std::shared_ptr<mkldnn::memory> user_memory_p,
194
194
const std::string& suffix, const std::vector<mkldnn::primitive>& pipeline,
195
- bool is_test = false ) { // NOLINT
195
+ bool is_persistent = false ) { // NOLINT
196
196
// create reorder primitive if the input format is not the preferred one
197
197
auto local_key = key_ + suffix;
198
198
auto key_reorder_p = key_ + suffix + " reorder_p" ;
@@ -213,7 +213,7 @@ class MKLDNNHandler {
213
213
pipeline.push_back (*reorder_p);
214
214
}
215
215
dev_ctx_.SetBlob (local_key, target_memory_p);
216
- } else if (!is_test ) {
216
+ } else if (!is_persistent ) {
217
217
// Make reorder if needed
218
218
auto reorder_p = std::static_pointer_cast<mkldnn::reorder>(
219
219
dev_ctx_.GetBlob (key_reorder_p));
You can’t perform that action at this time.
0 commit comments