Searched refs:conv_prim_desc (Results 1 – 1 of 1) sorted by relevance
1250 const ConvFwdPd& conv_prim_desc, in AllocateOutputTensor() argument1255 auto dst_pd = conv_prim_desc.dst_primitive_desc(); in AllocateOutputTensor()1306 const ConvFwdPd& conv_prim_desc, in AllocatePersistentTensor() argument1311 (conv_prim_desc.weights_primitive_desc().get_size() / sizeof(Tfilter))); in AllocatePersistentTensor()1319 sizeof(conv_prim_desc.weights_primitive_desc().desc().data.format) / in AllocatePersistentTensor()1325 conv_prim_desc.weights_primitive_desc().desc().data.format; in AllocatePersistentTensor()1329 const ConvFwdPd& conv_prim_desc, in AllocateFilterOutputTensor() argument1333 auto filter_pd = conv_prim_desc.weights_primitive_desc(); in AllocateFilterOutputTensor()1356 void PrepareAndExecuteNet(const ConvFwdPd& conv_prim_desc, in PrepareAndExecuteNet() argument1367 src->CheckReorderToOpMem(conv_prim_desc.src_primitive_desc()); in PrepareAndExecuteNet()[all …]