| 103 | #undef REGISTER_VARIANT_DECODE_TYPE |
| 104 | |
| 105 | UnaryVariantOpRegistry::AsyncVariantDeviceCopyFn* |
| 106 | UnaryVariantOpRegistry::GetDeviceCopyFn( |
| 107 | const VariantDeviceCopyDirection direction, const TypeIndex& type_index) { |
| 108 | auto found = device_copy_fns.find(std::make_pair(direction, type_index)); |
| 109 | if (found == device_copy_fns.end()) return nullptr; |
| 110 | return &found->second; |
| 111 | } |
| 112 | |
| 113 | void UnaryVariantOpRegistry::RegisterDeviceCopyFn( |
| 114 | const VariantDeviceCopyDirection direction, const TypeIndex& type_index, |