Searched refs:bias_mem (Results 1 – 2 of 2) sorted by relevance
109 context_.bias_mem->set_data_handle( in Execute()118 context_.bias_mem->set_data_handle( in Execute()128 context_.bias_mem->set_data_handle(DummyData); in Execute()143 std::shared_ptr<mkldnn::memory> bias_mem; member165 bias_mem(nullptr), in MklDnnMatMulFwdContext()271 context_.bias_mem.reset(new memory({{matmul_fwd_params.bias_dims}, in Setup()280 {MKLDNN_ARG_BIAS, *context_.bias_mem}, in Setup()
124 context_.bias_mem->set_data_handle( in Execute()135 context_.bias_mem->set_data_handle( in Execute()153 context_.bias_mem->set_data_handle(DummyData); in Execute()177 std::shared_ptr<mkldnn::memory> bias_mem; member199 bias_mem(nullptr), in ConvFwdContext()295 context_.bias_mem.reset(new memory( in Setup()302 {MKLDNN_ARG_BIAS, *context_.bias_mem}, in Setup()