Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.cc2792 Status MaybeAddAllReduceStartThunkToMap( in MaybeAddAllReduceStartThunkToMap() function
2844 TF_RETURN_IF_ERROR(MaybeAddAllReduceStartThunkToMap( in EmitNcclThunk()
2852 MaybeAddAllReduceStartThunkToMap(all_reduce_start_thunks_, op, nullptr)); in EmitNcclThunk()