Searched refs:conv_fwd_pd (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.cc | 1084 std::shared_ptr<ConvFwdPd> conv_fwd_pd = conv_fwd->GetPrimitiveDesc(); in Compute() local
1085 AllocateOutputTensor(context, *conv_fwd_pd, dst_dims_mkl_order, tf_fmt, in Compute()
1090 AllocateFilterOutputTensor(context, *conv_fwd_pd, in Compute()
1103 src.CheckReorderToOpMem(conv_fwd_pd.get()->src_primitive_desc()); in Compute()
1120 CacheFilter(context, conv_fwd_pd, filter_data, filter_tensor, in Compute()
1131 conv_fwd_pd.get()->weights_primitive_desc()); in Compute()
1134 conv_fwd_pd.get()->weights_primitive_desc(), in Compute()
1149 this->GetBiasHandle(context, conv_fwd_pd, bias_tensor); in Compute()
1404 const std::shared_ptr<ConvFwdPd>& conv_fwd_pd, in CacheFilter() argument
1419 filter.CheckReorderToOpMem(conv_fwd_pd.get()->weights_primitive_desc()); in CacheFilter()
[all …]
|