Searched defs:conv_prim_desc (Results 1 – 1 of 1) sorted by relevance
1085 const ConvFwdPd& conv_prim_desc, in AllocateOutputTensor()1220 void AllocateTensor(OpKernelContext* context, const ConvFwdPd& conv_prim_desc, in AllocateTensor()1249 void AllocateTensor(OpKernelContext* context, const ConvFwdPd& conv_prim_desc, in AllocateTensor()1255 const ConvFwdPd& conv_prim_desc, in AllocateFilterOutputTensor()1866 void AllocateTensor(OpKernelContext* context, const ConvFwdPd& conv_prim_desc, in AllocateTensor()2011 const ConvFwdPd& conv_prim_desc, in AllocateOutputTensor()