Searched refs:weight_mem (Results 1 – 1 of 1) sorted by relevance
107 context_.weight_mem->set_data_handle( in Execute()116 context_.weight_mem->set_data_handle( in Execute()127 context_.weight_mem->set_data_handle(DummyData); in Execute()142 std::shared_ptr<mkldnn::memory> weight_mem; member164 weight_mem(nullptr), in MklDnnMatMulFwdContext()267 context_.weight_mem.reset(new memory(context_.fwd_pd.get()->weights_desc(), in Setup()279 {MKLDNN_ARG_WEIGHTS, *context_.weight_mem}, in Setup()