Searched refs:StringToHloOpcode (Results 1 – 8 of 8) sorted by relevance
375 HloOpcode opcode = StringToHloOpcode(instr->opcode()).ValueOrDie(); in IsValueEffectiveInteger()399 HloOpcode opcode = StringToHloOpcode(proto->opcode()).ValueOrDie(); in IsInstructionOverLimit()476 TF_ASSIGN_OR_RETURN(HloOpcode opcode, StringToHloOpcode(root->opcode())); in AnalyzeConstantValueFallback()644 TF_ASSIGN_OR_RETURN(HloOpcode opcode, StringToHloOpcode(root->opcode())); in AnalyzeUpperBound()814 TF_ASSIGN_OR_RETURN(HloOpcode opcode, StringToHloOpcode(root->opcode())); in AnalyzeLowerBound()904 HloOpcode opcode = StringToHloOpcode(root->opcode()).ValueOrDie(); in AnalyzeConstant()1012 TF_ASSIGN_OR_RETURN(HloOpcode opcode, StringToHloOpcode(root->opcode())); in AnalyzeIsDynamic()1619 TF_ASSIGN_OR_RETURN(HloOpcode opcode, StringToHloOpcode(inst->opcode())); in CseOpHandle()1655 TF_ASSIGN_OR_RETURN(HloOpcode opcode, StringToHloOpcode(inst->opcode())); in SimplifyOp()1704 StringToHloOpcode(rhs_inst->opcode()).ValueOrDie(); in SimplifyOp()[all …]
90 HloOpcode opcode = StringToHloOpcode(instr_proto->opcode()).ValueOrDie(); in InstrIsSetBound()367 const HloOpcode opcode = StringToHloOpcode(instr.opcode()).ValueOrDie(); in IsConstantVisitor()3609 if (StringToHloOpcode(val_proto->opcode()).ValueOrDie() == in SetDimensionSizeInternal()
41 EXPECT_EQ(opcode, StringToHloOpcode(HloOpcodeString(opcode)).ValueOrDie()); in TEST()
34 StatusOr<HloOpcode> StringToHloOpcode(const std::string& opcode_name) { in StringToHloOpcode() function
184 StatusOr<HloOpcode> StringToHloOpcode(const std::string& opcode_name);
5487 auto status_or_result = StringToHloOpcode(val); in ParseOpcode()5495 status_or_result = StringToHloOpcode(wrapped_opcode); in ParseOpcode()
91 auto opcode_or = StringToHloOpcode(proto.opcode()); in CreateFromProto()
2067 assert(xla::StringToHloOpcode(get_tuple_element_proto->opcode()) in Lower()2079 assert(xla::StringToHloOpcode(xla_infeed_op_proto->opcode()) in Lower()2101 assert(xla::StringToHloOpcode(data_tuple_proto->opcode()) in Lower()