Searched refs:instr_proto (Results 1 – 1 of 1) sorted by relevance
141 bool InstrIsSetBound(const HloInstructionProto* instr_proto) { in InstrIsSetBound() argument142 HloOpcode opcode = StringToHloOpcode(instr_proto->opcode()).ValueOrDie(); in InstrIsSetBound()144 instr_proto->custom_call_target() == "SetBound") { in InstrIsSetBound()417 TF_ASSIGN_OR_RETURN(auto instr_proto, LookUpMutableInstruction(op)); in SetInstructionFrontendAttribute()418 auto* frontend_attributes = instr_proto->mutable_frontend_attributes(); in SetInstructionFrontendAttribute()3421 auto operand_is_constant = [&](const HloInstructionProto* instr_proto, in BuildDynamicInferenceGraph()3423 int64 operand_id = instr_proto->operand_ids(operand_index); in BuildDynamicInferenceGraph()3431 auto process_instruction = [&](const HloInstructionProto* instr_proto, in BuildDynamicInferenceGraph() argument3450 StringToHloOpcode(instr_proto->opcode())); in BuildDynamicInferenceGraph()3452 *new_instr = *instr_proto; in BuildDynamicInferenceGraph()[all …]