Searched defs:GetNcclCollectiveConfig (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ | ||
| H A D | lmhlo_to_jitrt.cc | 1148 static xla::gpu::NcclCollectiveConfig GetNcclCollectiveConfig( in GetNcclCollectiveConfig() function in tensorflow::CollectiveOpLowering |