Searched refs:root_opcode (Results 1 – 1 of 1) sorted by relevance
260 HloOpcode root_opcode = computation.root_instruction()->opcode(); in MaybeEmitDirectAtomicOperation() local269 root_opcode == HloOpcode::kParameter && in MaybeEmitDirectAtomicOperation()288 if (root_opcode == HloOpcode::kAdd) { in MaybeEmitDirectAtomicOperation()316 if (root_opcode == HloOpcode::kMaximum && is_atomic_integral) { in MaybeEmitDirectAtomicOperation()326 if (root_opcode == HloOpcode::kMinimum && is_atomic_integral) { in MaybeEmitDirectAtomicOperation()