Searched refs:HloOpcodeArity (Results 1 – 4 of 4) sorted by relevance
61 absl::optional<int> HloOpcodeArity(HloOpcode opcode) { in HloOpcodeArity() function
183 absl::optional<int> HloOpcodeArity(HloOpcode opcode);
93 absl::optional<int> arity = HloOpcodeArity(hlo->opcode()); in Preprocess()
136 absl::optional<int> arity = HloOpcodeArity(opcode); in CreateFromProto()