Home
last modified time | relevance | path

Searched refs:filter_mem (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc113 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; member
169 filter_mem(nullptr), in ConvBwdInputContext()
221 context_.filter_mem.reset(new memory( in Setup()
231 {MKLDNN_ARG_WEIGHTS, *context_.filter_mem}, in Setup()
Dmkl_conv_ops.cc121 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; member
198 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()