Searched refs:EmitAtomicOperationForNestedComputation (Results 1 – 3 of 3) sorted by relevance
149 Status EmitAtomicOperationForNestedComputation(
449 Status IrEmitter::EmitAtomicOperationForNestedComputation( in EmitAtomicOperationForNestedComputation() function in xla::gpu::IrEmitter
2139 return EmitAtomicOperationForNestedComputation( in EmitSelectAndScatter()2423 return EmitAtomicOperationForNestedComputation( in EmitScatter()3887 TF_CHECK_OK(EmitAtomicOperationForNestedComputation( in EmitReductionOutputForRowReduction()4047 TF_CHECK_OK(EmitAtomicOperationForNestedComputation( in EmitReductionOutputForColumnReduction()