Searched refs:EmitAtomicOperationForNestedComputation (Results 1 – 3 of 3) sorted by relevance
160 Status EmitAtomicOperationForNestedComputation(
497 Status IrEmitter::EmitAtomicOperationForNestedComputation( in EmitAtomicOperationForNestedComputation() function in xla::gpu::IrEmitter
2413 return EmitAtomicOperationForNestedComputation( in EmitSelectAndScatterFromMlir()2710 return EmitAtomicOperationForNestedComputation( in EmitScatter()4516 TF_CHECK_OK(EmitAtomicOperationForNestedComputation( in EmitEpilogueForReduction()4556 TF_CHECK_OK(EmitAtomicOperationForNestedComputation( in EmitEpilogueForReduction()