Home
last modified time | relevance | path

Searched refs:diff_dst_mem (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_pooling_ops_common.cc166 context_.diff_dst_mem.reset(new memory(context_.bwd_pd.get()->diff_dst_desc(), in Setup()
174 {{MKLDNN_ARG_DIFF_DST, *context_.diff_dst_mem}, in Setup()
180 {{MKLDNN_ARG_DIFF_DST, *context_.diff_dst_mem}, in Setup()
192 context_.diff_dst_mem->set_data_handle( in Execute()
201 context_.diff_dst_mem->set_data_handle( in Execute()
213 context_.diff_dst_mem->set_data_handle(DummyData); in Execute()
Dmkl_conv_grad_input_ops.cc115 context_.diff_dst_mem->set_data_handle( in Execute()
122 context_.diff_dst_mem->set_data_handle( in Execute()
131 context_.diff_dst_mem->set_data_handle(DummyData); in Execute()
145 std::shared_ptr<mkldnn::memory> diff_dst_mem; member
170 diff_dst_mem(nullptr), in ConvBwdInputContext()
223 context_.diff_dst_mem.reset(new memory( in Setup()
230 {{MKLDNN_ARG_DIFF_DST, *context_.diff_dst_mem}, in Setup()
Dmkl_conv_grad_filter_ops.cc119 context_.diff_dst_mem->set_data_handle( in Execute()
130 context_.diff_dst_mem->set_data_handle( in Execute()
141 context_.diff_dst_mem->set_data_handle(DummyData); in Execute()
166 std::shared_ptr<mkldnn::memory> diff_dst_mem; member
194 diff_dst_mem(nullptr), in ConvBwdFilterContext()
263 context_.diff_dst_mem.reset( in Setup()
279 {MKLDNN_ARG_DIFF_DST, *context_.diff_dst_mem}}); in Setup()
286 {MKLDNN_ARG_DIFF_DST, *context_.diff_dst_mem}}); in Setup()
Dmkl_relu_op.cc261 context_.diff_dst_mem->set_data_handle( in Execute()
268 context_.diff_dst_mem->set_data_handle( in Execute()
279 context_.diff_dst_mem->set_data_handle(DummyData); in Execute()
290 std::shared_ptr<memory> diff_dst_mem; member
321 diff_dst_mem(nullptr), in EltwiseBwdContext()
358 context_.diff_dst_mem.reset( in Setup()
366 {MKLDNN_ARG_DIFF_DST, *context_.diff_dst_mem}, in Setup()
Dmkl_fused_batch_norm_op.cc433 context_.diff_dst_mem->set_data_handle( in Execute()
452 context_.diff_dst_mem->set_data_handle( in Execute()
472 context_.diff_dst_mem->set_data_handle(DummyData); in Execute()
495 std::shared_ptr<mkldnn::memory> diff_dst_mem; member
513 diff_dst_mem(nullptr), in BatchNormBwdContext()
557 context_.diff_dst_mem.reset( in Setup()
573 {MKLDNN_ARG_DIFF_DST, *context_.diff_dst_mem}, in Setup()
Dmkl_pooling_ops_common.h260 std::shared_ptr<mkldnn::memory> diff_dst_mem; member
285 diff_dst_mem(nullptr), in PoolingBwdContext()