Searched refs:conv1_src_memory (Results 1 – 1 of 1) sorted by relevance
120 auto conv1_src_memory = user_src_memory; in MKLConvImpl() local123 conv1_src_memory = memory(conv1_prim_desc.src_primitive_desc()); in MKLConvImpl()124 net.push_back(reorder(user_src_memory, conv1_src_memory)); in MKLConvImpl()144 net.push_back(convolution_forward(conv1_prim_desc, conv1_src_memory, in MKLConvImpl()