Searched refs:conv_fwd_pd (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.cc | 665 std::shared_ptr<ConvFwdPd> conv_fwd_pd = conv_fwd->GetPrimitiveDesc(); in Compute() local
666 AllocateOutputTensor(context, *conv_fwd_pd, dst_dims_mkl_order, tf_fmt, in Compute()
671 AllocateFilterOutputTensor(context, *conv_fwd_pd, in Compute()
681 if (src_md != conv_fwd_pd->src_desc()) { in Compute()
683 src.CheckReorderToOpMem(conv_fwd_pd->src_desc(), cpu_engine_, context); in Compute()
691 if (filter_md != conv_fwd_pd->weights_desc()) { in Compute()
700 CacheFilter(context, conv_fwd_pd, filter_data, filter_tensor, in Compute()
703 filter_data = GetCachedFilter(context, conv_fwd_pd->weights_desc()); in Compute()
709 filter.CheckReorderToOpMem(conv_fwd_pd->weights_desc(), cpu_engine_, in Compute()
713 conv_fwd_pd->weights_desc(), in Compute()
[all …]
|