Home
last modified time | relevance | path

Searched refs:diff_weights_desc (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_types.h61 diff_filter_md != op_pd->diff_weights_desc()
106 #define PRIMITIVE_DESC_DIFF_WEIGHTS diff_weights_desc()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_filter_ops.cc262 new memory(bwd_filter_pd->diff_weights_desc(), cpu_engine_, DummyData)); in Setup()
572 if (diff_filter_md != bwd_filter_pd->diff_weights_desc()) { in Compute()
578 bwd_filter_pd->diff_weights_desc()); in Compute()
Dmkl_fused_batch_norm_op.cc534 auto diff_weights_desc = weights_desc; in Setup() local
565 new memory(diff_weights_desc, cpu_engine_, DummyData)); in Setup()