Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc161 std::shared_ptr<memory::desc> diff_dst_md; member
411 auto diff_dst_md = in Compute() local
Dmkl_avgpooling_op.cc234 memory::desc diff_dst_md = in Compute() local
Dmkl_conv_grad_filter_ops.cc183 std::shared_ptr<mkldnn::memory::desc> diff_dst_md; member
448 auto diff_dst_md = in Compute() local
Dmkl_maxpooling_op.cc291 memory::desc diff_dst_md = in Compute() local
Dmkl_fused_batch_norm_op.cc385 memory::desc diff_dst_md; member
527 auto diff_dst_md = bwdParams.diff_dst_md; in Setup() local
1176 memory::desc diff_dst_md = in Compute() local
Dmkl_relu_op.cc298 std::shared_ptr<memory::desc> diff_dst_md; member
623 memory::desc diff_dst_md({}, memory::data_type::undef, in Compute() local
/external/tensorflow/tensorflow/core/util/
Dmkl_types.h58 #define IS_DIFF_DST_REORDER_NEEDED(diff_dst_md, op_pd, op) \ argument