We use a vector instead of a map since we expect there to be very few registrations.
| 42 | // We use a vector instead of a map since we expect there to be very |
| 43 | // few registrations. |
| 44 | std::vector<RegistrationInfo>* MutableRegistry() { |
| 45 | static std::vector<RegistrationInfo>* registry = |
| 46 | new std::vector<RegistrationInfo>; |
| 47 | return registry; |
| 48 | } |
| 49 | |
| 50 | void CopyHostToDevice(const Tensor* input, Allocator* cpu_allocator, |
| 51 | Allocator* out_allocator, StringPiece edge_name, |