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

code style change.

Browse Source
This commit is contained in:
Li, Guizi 2019-11-07 09:06:26 +08:00
parent 355a233f79
commit eac539e574
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/core/kernels/mkl_conv_ops.cc
View File

@ -1625,15 +1625,15 @@ class MklQuantizedConv2DOp
input_bias_->set_data_handle(bias_buf); input_bias_->set_data_handle(bias_buf);
} }
if (!scaled_bias_buf) if (!scaled_bias_buf_)
AllocTmpBuffer<Tbias>(context, &scaled_bias_tensor_, AllocTmpBuffer<Tbias>(context, &scaled_bias_tensor_,
conv_fwd_pd->bias_primitive_desc(), conv_fwd_pd->bias_primitive_desc(),
&scaled_bias_buf); &scaled_bias_buf_);
if (!scaled_bias_) { if (!scaled_bias_) {
scaled_bias_ = scaled_bias_ =
new MEMORY_CONSTRUCTOR(bias_md, this->cpu_engine_, scaled_bias_buf); new MEMORY_CONSTRUCTOR(bias_md, this->cpu_engine_, scaled_bias_buf_);
} else { } else {
scaled_bias_->set_data_handle(scaled_bias_buf); scaled_bias_->set_data_handle(scaled_bias_buf_);
} }
auto reorder_desc = REORDER_PD_CONSTRUCTOR_WITH_ATTR( auto reorder_desc = REORDER_PD_CONSTRUCTOR_WITH_ATTR(
input_bias_->GET_DESC, scaled_bias_->GET_DESC, this->cpu_engine_, input_bias_->GET_DESC, scaled_bias_->GET_DESC, this->cpu_engine_,
@ -1658,7 +1658,7 @@ class MklQuantizedConv2DOp
memory* scaled_bias_ = nullptr; memory* scaled_bias_ = nullptr;
Tensor scaled_bias_tensor_; Tensor scaled_bias_tensor_;
void* scaled_bias_buf = nullptr; void* scaled_bias_buf_ = nullptr;
private: private:
std::vector<float> scales_; std::vector<float> scales_;