Home
last modified time | relevance | path

Searched refs:root_opcode (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc260 HloOpcode root_opcode = computation.root_instruction()->opcode(); in MaybeEmitDirectAtomicOperation() local
269 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()