Searched refs:filter_mem (Results 1 – 2 of 2) sorted by relevance
102 context_.filter_mem->set_data_handle( in Execute()111 context_.filter_mem->set_data_handle(DummyData); in Execute()136 std::shared_ptr<mkldnn::memory> filter_mem; member162 filter_mem(nullptr), in ConvBwdInputContext()209 context_.filter_mem.reset(new memory( in Setup()228 *context_.bwd_input_pd, *context_.diff_dst_mem, *context_.filter_mem, in Setup()
124 context_.filter_mem->set_data_handle( in Execute()134 context_.filter_mem->set_data_handle(DummyData); in Execute()149 context_.filter_mem->set_data_handle( in Execute()157 context_.filter_mem->set_data_handle(DummyData); in Execute()178 std::shared_ptr<mkldnn::memory> filter_mem; member202 filter_mem(nullptr), in ConvFwdContext()291 context_.filter_mem.reset( in Setup()303 *context_.fwd_pd, *context_.src_mem, *context_.filter_mem, in Setup()308 *context_.filter_mem, *context_.dst_mem)); in Setup()