Searched refs:filter_mem (Results 1 – 2 of 2) sorted by relevance
113 context_.filter_mem->set_data_handle( in Execute()120 context_.filter_mem->set_data_handle( in Execute()130 context_.filter_mem->set_data_handle(DummyData); in Execute()144 std::shared_ptr<mkldnn::memory> filter_mem; member169 filter_mem(nullptr), in ConvBwdInputContext()221 context_.filter_mem.reset(new memory( in Setup()231 {MKLDNN_ARG_WEIGHTS, *context_.filter_mem}, in Setup()
121 context_.filter_mem->set_data_handle( in Execute()132 context_.filter_mem->set_data_handle( in Execute()151 context_.filter_mem->set_data_handle(DummyData); in Execute()176 std::shared_ptr<mkldnn::memory> filter_mem; member198 filter_mem(nullptr), in ConvFwdContext()288 context_.filter_mem.reset(new memory(context_.fwd_pd.get()->weights_desc(), in Setup()301 {MKLDNN_ARG_WEIGHTS, *context_.filter_mem}, in Setup()308 {MKLDNN_ARG_WEIGHTS, *context_.filter_mem}, in Setup()