| 21 | namespace xla { |
| 22 | |
| 23 | string HloOpcodeString(HloOpcode opcode) { |
| 24 | switch (opcode) { |
| 25 | #define CASE_OPCODE_STRING(enum_name, opcode_name, ...) \ |
| 26 | case HloOpcode::enum_name: \ |
| 27 | return opcode_name; |
| 28 | HLO_OPCODE_LIST(CASE_OPCODE_STRING) |
| 29 | #undef CASE_OPCODE_STRING |
| 30 | } |
| 31 | } |
| 32 | |
| 33 | StatusOr<HloOpcode> StringToHloOpcode(const string& opcode_name) { |
| 34 | static auto* opcode_map = new absl::flat_hash_map<string, HloOpcode>({ |
no outgoing calls