| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_avgpooling_op.cc | 635 memory::desc diff_dst_md = variable
642 if (diff_dst_md.data.format != pooling_bwd->GetDiffDstFormat()) {
646 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor);
|
| mkl_conv_grad_filter_ops.cc | 183 std::shared_ptr<mkldnn::memory::desc> diff_dst_md; member in struct:tensorflow::MklConvBwdFilterPrimitive::ConvBwdFilterContext
203 diff_dst_md(nullptr),
213 context_.diff_dst_md.reset(
230 *context_.diff_bias_md, *context_.diff_dst_md,
237 *context_.diff_dst_md, convBwdFilterDims.strides,
245 *context_.diff_filter_md, *context_.diff_dst_md,
450 auto diff_dst_md =
548 if (diff_dst_md.data.format !=
550 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor);
|
| mkl_conv_grad_input_ops.cc | 152 std::shared_ptr<memory::desc> diff_dst_md; member in struct:tensorflow::MklConvBwdInputPrimitive::ConvBwdInputContext
171 diff_dst_md(nullptr),
182 context_.diff_dst_md.reset(
189 *context_.diff_dst_md, convBwdInputDims.strides,
195 *context_.filter_md, *context_.diff_dst_md, convBwdInputDims.strides,
395 auto diff_dst_md =
451 if (diff_dst_md.data.format != conv_bwd_input->GetDiffDstMemoryFormat()) {
452 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor);
|
| mkl_maxpooling_op.cc | 746 memory::desc diff_dst_md = variable
753 if (diff_dst_md.data.format != pooling_bwd->GetDiffDstFormat()) {
757 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor);
[all...] |
| mkl_pooling_ops_common.h | 269 std::shared_ptr<mkldnn::memory::desc> diff_dst_md; member in struct:tensorflow::MklPoolingBwdPrimitive::PoolingBwdContext
291 diff_dst_md(nullptr),
|
| mkl_relu_op.cc | 303 std::shared_ptr<memory::desc> diff_dst_md; member in struct:tensorflow::MklEltwiseBwdPrimitive::EltwiseBwdContext
328 diff_dst_md(nullptr),
343 context_.diff_dst_md.reset(new memory::desc(bwdParams.common_md.data));
348 new memory::primitive_desc(*context_.diff_dst_md, cpu_engine_));
357 bwdParams.alg_kind, *context_.diff_dst_md, *context_.src_md,
588 memory::desc diff_dst_md({}, memory::data_undef, memory::format_undef);
593 diff_dst_md = src_md;
604 diff_dst_md =
608 diff_dst_md = dnn_shape_diff_dst.GetMklLayout();
624 diff_dst_md = dnn_shape_diff_dst.GetMklLayout()
[all...] |
| mkl_fused_batch_norm_op.cc | 390 auto diff_dst_md = local
416 prop_kind::backward, diff_dst_md, src_md, bwdParams.eps,
425 new memory({diff_dst_md, cpu_engine_}, DummyData));
939 memory::desc diff_dst_md = variable
[all...] |
