Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc263 bool is_atomic_integral = element_type == S32 || element_type == U32 || in MaybeEmitDirectAtomicOperation() local
270 (element_type == F32 || is_atomic_integral) && in MaybeEmitDirectAtomicOperation()
307 if (is_atomic_integral) { in MaybeEmitDirectAtomicOperation()
316 if (root_opcode == HloOpcode::kMaximum && is_atomic_integral) { in MaybeEmitDirectAtomicOperation()
326 if (root_opcode == HloOpcode::kMinimum && is_atomic_integral) { in MaybeEmitDirectAtomicOperation()