Home
last modified time | relevance | path

Searched defs:diff_dst_md (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc151 std::shared_ptr<memory::desc> diff_dst_md; member
412 auto diff_dst_md = in Compute() local
Dmkl_avgpooling_op.cc244 memory::desc diff_dst_md = in Compute() local
Dmkl_conv_grad_filter_ops.cc173 std::shared_ptr<dnnl::memory::desc> diff_dst_md; member
452 auto diff_dst_md = in Compute() local
Dmkl_maxpooling_op.cc297 memory::desc diff_dst_md = in Compute() local
Dmkl_fused_batch_norm_op.cc399 memory::desc diff_dst_md; member
546 auto diff_dst_md = bwdParams.diff_dst_md; in Setup() local
1267 memory::desc diff_dst_md = in Compute() local
Dmkl_relu_op.cc305 std::shared_ptr<memory::desc> diff_dst_md; member
625 memory::desc diff_dst_md({}, memory::data_type::undef, in Compute() local