Searched refs:root_opcode (Results 1 – 1 of 1) sorted by relevance
185 HloOpcode root_opcode = computation.root_instruction()->opcode(); in MaybeEmitDirectAtomicOperation() local193 if (root_opcode == HloOpcode::kCopy && in MaybeEmitDirectAtomicOperation()206 if (root_opcode == HloOpcode::kAdd) { in MaybeEmitDirectAtomicOperation()224 if (root_opcode == HloOpcode::kMaximum && is_atomic_integral) { in MaybeEmitDirectAtomicOperation()234 if (root_opcode == HloOpcode::kMinimum && is_atomic_integral) { in MaybeEmitDirectAtomicOperation()