Special casing UnaryOpFn per op and per device.
| 173 | |
| 174 | // Special casing UnaryOpFn per op and per device. |
| 175 | UnaryVariantOpRegistry::VariantUnaryOpFn* UnaryVariantOpRegistry::GetUnaryOpFn( |
| 176 | VariantUnaryOp op, StringPiece device, const TypeIndex& type_index) { |
| 177 | auto found = unary_op_fns.find({op, device, type_index}); |
| 178 | if (found == unary_op_fns.end()) return nullptr; |
| 179 | return &found->second; |
| 180 | } |
| 181 | |
| 182 | void UnaryVariantOpRegistry::RegisterUnaryOpFn( |
| 183 | VariantUnaryOp op, const string& device, const TypeIndex& type_index, |