Searched refs:bias_md (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 153 std::shared_ptr<mkldnn::memory::desc> bias_md; member
171 bias_md(nullptr), in MklDnnMatMulFwdContext()
191 context_.bias_md.reset(new memory::desc({matmul_fwd_params.bias_dims}, in Setup()
197 *context_.bias_md, *context_.dst_md)); in Setup()
|
| D | mkl_conv_ops.cc | 186 std::shared_ptr<mkldnn::memory::desc> bias_md; member
204 bias_md(nullptr), in ConvFwdContext()
228 context_.bias_md.reset(new memory::desc({convFwdDims.bias_dims}, in Setup()
236 *context_.src_md, *context_.filter_md, *context_.bias_md, in Setup()
1550 auto bias_md = memory::desc({static_cast<int>(bias_tensor.NumElements())}, in GetBiasHandle() local
1555 input_bias_ = new memory(bias_md, this->cpu_engine_, bias_buf); in GetBiasHandle()
1564 scaled_bias_ = new memory(bias_md, this->cpu_engine_, scaled_bias_buf_); in GetBiasHandle()
|