Searched refs:conv1_prim_desc (Results 1 – 1 of 1) sorted by relevance
113 auto conv1_prim_desc = in MKLConvImpl() local119 if (memory::primitive_desc(conv1_prim_desc.src_primitive_desc()) != in MKLConvImpl()121 conv1_src_memory = memory(conv1_prim_desc.src_primitive_desc()); in MKLConvImpl()126 if (memory::primitive_desc(conv1_prim_desc.weights_primitive_desc()) != in MKLConvImpl()128 conv1_weights_memory = memory(conv1_prim_desc.weights_primitive_desc()); in MKLConvImpl()135 (memory::primitive_desc(conv1_prim_desc.dst_primitive_desc()) != in MKLConvImpl()138 ? memory(conv1_prim_desc.dst_primitive_desc()) in MKLConvImpl()142 net.push_back(convolution_forward(conv1_prim_desc, conv1_src_memory, in MKLConvImpl()