Home
last modified time | relevance | path

Searched refs:diff_src_mem (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_pooling_ops_common.cc164 context_.diff_src_mem.reset(new memory(context_.bwd_pd.get()->diff_src_desc(), in Setup()
176 {MKLDNN_ARG_DIFF_SRC, *context_.diff_src_mem}}); in Setup()
181 {MKLDNN_ARG_DIFF_SRC, *context_.diff_src_mem}}); in Setup()
194 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data), in Execute()
203 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data)); in Execute()
214 context_.diff_src_mem->set_data_handle(DummyData); in Execute()
Dmkl_conv_grad_input_ops.cc111 context_.diff_src_mem->set_data_handle( in Execute()
118 context_.diff_src_mem->set_data_handle( in Execute()
129 context_.diff_src_mem->set_data_handle(DummyData); in Execute()
143 std::shared_ptr<mkldnn::memory> diff_src_mem; member
168 : diff_src_mem(nullptr), in ConvBwdInputContext()
219 context_.diff_src_mem.reset(new memory( in Setup()
232 {MKLDNN_ARG_DIFF_SRC, *context_.diff_src_mem}}); in Setup()
Dmkl_relu_op.cc263 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data), in Execute()
270 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data)); in Execute()
280 context_.diff_src_mem->set_data_handle(DummyData); in Execute()
291 std::shared_ptr<memory> diff_src_mem; member
322 diff_src_mem(nullptr), in EltwiseBwdContext()
360 context_.diff_src_mem.reset( in Setup()
367 {MKLDNN_ARG_DIFF_SRC, *context_.diff_src_mem}}); in Setup()
Dmkl_fused_batch_norm_op.cc443 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data), in Execute()
462 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data)); in Execute()
477 context_.diff_src_mem->set_data_handle(DummyData); in Execute()
484 memory::desc GetDiffSrcPd() { return context_.diff_src_mem->get_desc(); } in GetDiffSrcPd()
498 std::shared_ptr<mkldnn::memory> diff_src_mem; member
516 diff_src_mem(nullptr) {} in BatchNormBwdContext()
566 context_.diff_src_mem.reset(new memory(src_md, cpu_engine_, DummyData)); in Setup()
575 {MKLDNN_ARG_DIFF_SRC, *context_.diff_src_mem}, in Setup()
Dmkl_pooling_ops_common.h259 std::shared_ptr<mkldnn::memory> diff_src_mem; member
284 diff_src_mem(nullptr), in PoolingBwdContext()