HomeSort by relevance Sort by last modified time
    Searched defs:target_diff_dst_md (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_avgpooling_op.cc 587 memory::desc target_diff_dst_md = this->ConfigureInputGradient( variable
597 static_cast<memory::format>(target_diff_dst_md.data.format));
616 output_diff_src.GetUsrMemDesc(), target_diff_dst_md,
634 memory::primitive_desc(target_diff_dst_md, cpu_engine));
mkl_pooling_ops_common.h 392 memory::desc target_diff_dst_md = local
395 return grad_reorder_needed ? target_diff_dst_md : original_input_grad_md;
mkl_maxpooling_op.cc 657 memory::desc target_diff_dst_md = this->ConfigureInputGradient( variable
679 target_diff_dst_md,
700 memory::primitive_desc(target_diff_dst_md, cpu_engine),
    [all...]

Completed in 152 milliseconds