Searched refs:scaled_bias_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_qmatmul_op.cc | 122 if (this->scaled_bias_ != nullptr) { in ~MklDnnQuantizedMatMulOp()
123 delete this->scaled_bias_; in ~MklDnnQuantizedMatMulOp()
124 scaled_bias_ = nullptr; in ~MklDnnQuantizedMatMulOp()
480 scaled_bias_ = in GetBiasHandle()
484 *input_bias_, *scaled_bias_, bias_attr); in GetBiasHandle()
487 {MKLDNN_ARG_FROM, *input_bias_}, {MKLDNN_ARG_TO, *scaled_bias_}}; in GetBiasHandle()
490 return reinterpret_cast<Tbias*>(scaled_bias_->get_data_handle()); in GetBiasHandle()
502 memory* scaled_bias_ = nullptr; member in tensorflow::MklDnnQuantizedMatMulOp
|
| D | mkl_conv_ops.cc | 1353 if (this->scaled_bias_ != nullptr) { in ~MklQuantizedConv2DOp()
1354 delete this->scaled_bias_; in ~MklQuantizedConv2DOp()
1355 scaled_bias_ = nullptr; in ~MklQuantizedConv2DOp()
1563 if (!scaled_bias_) { in GetBiasHandle()
1564 scaled_bias_ = new memory(bias_md, this->cpu_engine_, scaled_bias_buf_); in GetBiasHandle()
1566 scaled_bias_->set_data_handle(scaled_bias_buf_); in GetBiasHandle()
1570 this->cpu_engine_, scaled_bias_->get_desc(), bias_attr); in GetBiasHandle()
1571 CreateAndExecuteReorder(reorder_desc, *input_bias_, *scaled_bias_, in GetBiasHandle()
1575 reinterpret_cast<Tbias*>(scaled_bias_->get_data_handle()); in GetBiasHandle()
1577 CacheBias(context, conv_fwd_pd, bias_data, scaled_bias_); in GetBiasHandle()
[all …]
|