Searched refs:conv_prim_desc (Results 1 – 1 of 1) sorted by relevance
849 const ConvFwdPd& conv_prim_desc, in AllocateOutputTensor() argument855 auto dst_md = conv_prim_desc.dst_desc(); in AllocateOutputTensor()967 const ConvFwdPd& conv_prim_desc) const { in GetFilterTfDataFormat()975 const ConvFwdPd& conv_prim_desc, in AllocatePersistentTensor() argument981 (conv_prim_desc.weights_desc().get_size() / sizeof(Tfilter))); in AllocatePersistentTensor()991 memory::desc weights_desc = conv_prim_desc.weights_desc(); in AllocatePersistentTensor()1004 const ConvFwdPd& conv_prim_desc, in AllocatePersistentTensor() argument1006 AllocatePersistentTensor(context, conv_prim_desc, filter_tensor, nullptr); in AllocatePersistentTensor()1010 const ConvFwdPd& conv_prim_desc, in AllocateFilterOutputTensor() argument1014 auto filter_md = conv_prim_desc.weights_desc(); in AllocateFilterOutputTensor()[all …]