Searched refs:diff_dst_md (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_grad_input_ops.cc | 161 std::shared_ptr<memory::desc> diff_dst_md; member
178 diff_dst_md(nullptr) {} in ConvBwdInputContext()
192 context_.diff_dst_md.reset(new memory::desc( in Setup()
203 *context_.filter_md, *context_.diff_dst_md, convBwdInputDims.strides, in Setup()
209 *context_.diff_src_md, *context_.filter_md, *context_.diff_dst_md, in Setup()
411 auto diff_dst_md = in Compute() local
474 if (diff_dst_md != bwd_input_pd->diff_dst_desc()) { in Compute()
475 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 183 std::shared_ptr<mkldnn::memory::desc> diff_dst_md; member
201 diff_dst_md(nullptr) {} in ConvBwdFilterContext()
218 context_.diff_dst_md.reset(new memory::desc( in Setup()
233 *context_.src_md, *context_.diff_filter_md, *context_.diff_dst_md, in Setup()
243 *context_.diff_dst_md, convBwdFilterDims.strides, in Setup()
249 *context_.diff_filter_md, *context_.diff_dst_md, in Setup()
448 auto diff_dst_md = in Compute() local
552 if (diff_dst_md != bwd_filter_pd->diff_dst_desc()) { in Compute()
553 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
|
| D | mkl_relu_op.cc | 298 std::shared_ptr<memory::desc> diff_dst_md; member
324 diff_dst_md(nullptr), in EltwiseBwdContext()
338 context_.diff_dst_md.reset(new memory::desc(bwdParams.common_md.data)); in Setup()
340 context_.diff_dst_mpd.reset(new memory::desc(*context_.diff_dst_md)); in Setup()
348 bwdParams.alg_kind, *context_.diff_dst_md, *context_.src_md, in Setup()
623 memory::desc diff_dst_md({}, memory::data_type::undef, in Compute() local
629 diff_dst_md = src_md; in Compute()
640 diff_dst_md = memory::desc( in Compute()
645 diff_dst_md = dnn_shape_diff_dst.GetMklLayout(); in Compute()
662 diff_dst_md = dnn_shape_diff_dst.GetMklLayout(); in Compute()
[all …]
|
| D | mkl_fused_batch_norm_op.cc | 385 memory::desc diff_dst_md; member
389 memory::desc src_md, memory::desc diff_dst_md) in MklBatchNormBwdParams()
396 diff_dst_md(diff_dst_md) {} in MklBatchNormBwdParams()
527 auto diff_dst_md = bwdParams.diff_dst_md; in Setup() local
552 prop_kind::backward, diff_dst_md, src_md, bwdParams.eps, bn_flags); in Setup()
558 new memory(diff_dst_md, cpu_engine_, DummyData)); in Setup()
1176 memory::desc diff_dst_md = in Compute() local
1194 reorder_diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
1196 diff_dst_md = src_md; in Compute()
1202 reorder_src.CheckReorderToOpMem(diff_dst_md, cpu_engine_, context); in Compute()
[all …]
|
| D | mkl_avgpooling_op.cc | 234 memory::desc diff_dst_md = in Compute() local
266 (diff_dst_md != pooling_bwd_pd->diff_dst_desc())) { in Compute()
267 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor); in Compute()
|
| D | mkl_maxpooling_op.cc | 291 memory::desc diff_dst_md = in Compute() local
319 (diff_dst_md != pooling_bwd_pd->diff_dst_desc())) { in Compute()
320 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_types.h | 58 #define IS_DIFF_DST_REORDER_NEEDED(diff_dst_md, op_pd, op) \ argument
59 diff_dst_md != op_pd->diff_dst_desc()
|