Searched refs:input_bias_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_qmatmul_op.cc | 118 if (this->input_bias_ != nullptr) { in ~MklDnnQuantizedMatMulOp()
119 delete this->input_bias_; in ~MklDnnQuantizedMatMulOp()
120 input_bias_ = nullptr; in ~MklDnnQuantizedMatMulOp()
478 input_bias_ = new memory(mkldnn_matmul_fwd_pd->bias_desc(), in GetBiasHandle()
484 *input_bias_, *scaled_bias_, bias_attr); in GetBiasHandle()
487 {MKLDNN_ARG_FROM, *input_bias_}, {MKLDNN_ARG_TO, *scaled_bias_}}; in GetBiasHandle()
501 memory* input_bias_ = nullptr; member in tensorflow::MklDnnQuantizedMatMulOp
|
| D | mkl_conv_ops.cc | 1348 if (this->input_bias_ != nullptr) { in ~MklQuantizedConv2DOp()
1349 delete this->input_bias_; in ~MklQuantizedConv2DOp()
1350 input_bias_ = nullptr; in ~MklQuantizedConv2DOp()
1554 if (!input_bias_) { in GetBiasHandle()
1555 input_bias_ = new memory(bias_md, this->cpu_engine_, bias_buf); in GetBiasHandle()
1557 input_bias_->set_data_handle(bias_buf); in GetBiasHandle()
1569 ReorderPd(this->cpu_engine_, input_bias_->get_desc(), in GetBiasHandle()
1571 CreateAndExecuteReorder(reorder_desc, *input_bias_, *scaled_bias_, in GetBiasHandle()
1587 memory* input_bias_ = nullptr; member in tensorflow::MklQuantizedConv2DOp
|