Searched refs:custom_call_instruction (Results 1 – 1 of 1) sorted by relevance
584 void SetCustomCallInstruction(HloInstruction* custom_call_instruction) { in SetCustomCallInstruction() argument586 custom_call_instruction_ = custom_call_instruction; in SetCustomCallInstruction()587 is_custom_call_computation_ |= (custom_call_instruction != nullptr); in SetCustomCallInstruction()