MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / HloOpcodeString

Function HloOpcodeString

tensorflow/compiler/xla/service/hlo_opcode.cc:23–31  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

21namespace xla {
22
23string 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
33StatusOr<HloOpcode> StringToHloOpcode(const string& opcode_name) {
34 static auto* opcode_map = new absl::flat_hash_map<string, HloOpcode>({

Callers 15

GetXfeedShapeFunction · 0.85
DefaultActionMethod · 0.85
DoInfoCommandFunction · 0.85
GetProgramShapeMethod · 0.85
SetDynamicBindingMethod · 0.85
BinaryOpMethod · 0.85
BuildConstantSubGraphMethod · 0.85
AddInstructionMethod · 0.85
CheckOperandCountFunction · 0.85
HandleSortMethod · 0.85
HandleReduceMethod · 0.85
VerifyAsynchronousCopiesFunction · 0.85

Calls

no outgoing calls

Tested by 8

TEST_FFunction · 0.68
TESTFunction · 0.68
MakeReductionFunction · 0.68
DefaultActionMethod · 0.68
FindHloInstructionFunction · 0.68
NameMethod · 0.68
NameMethod · 0.68