Home
last modified time | relevance | path

Searched defs:conv_fwd_pd (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/
DConv.cpp204 auto conv_fwd_pd = dnnl::convolution_forward::primitive_desc( in convolution() local
292 auto conv_fwd_pd = dnnl::convolution_forward::primitive_desc( in convolution_backward_weights() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.cc825 std::shared_ptr<ConvFwdPd> conv_fwd_pd = conv_fwd->GetPrimitiveDesc(); in Compute() local
1295 const std::shared_ptr<ConvFwdPd>& conv_fwd_pd, in CacheFilter()
1770 std::shared_ptr<ConvFwdPd>& conv_fwd_pd, in GetBiasHandle()
1890 const std::shared_ptr<ConvFwdPd>& conv_fwd_pd, in CacheBias()