| 131 | static const int kSmallBatchSize = 32; |
| 132 | |
| 133 | inline void execute_primitives( |
| 134 | std::vector<dnnl::primitive>& primitives, std::shared_ptr<stream> stream, |
| 135 | std::vector<std::unordered_map<int, memory>>& net_args) { |
| 136 | DCHECK_EQ(primitives.size(), net_args.size()); |
| 137 | for (size_t i = 0; i < primitives.size(); ++i) { |
| 138 | primitives.at(i).execute(*stream, net_args.at(i)); |
| 139 | } |
| 140 | } |
| 141 | |
| 142 | // In OneDNN v1.x, the format (ex. NCHW) used to initialize a memory descriptor |
| 143 | // (md) structure will no longer be recorded in its `format` field. Instead, it |