Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dnccl_collective_permute_thunk.h35 struct SourceTargetMapEntry { struct
41 absl::flat_hash_map<int64_t, SourceTargetMapEntry>;
48 static SourceTargetMapEntry GetSourceTarget( in GetSourceTarget()
52 return SourceTargetMapEntry{}; in GetSourceTarget()
96 NcclCollectivePermuteConfig::SourceTargetMapEntry source_target,
H A Dnccl_collective_permute_thunk.cc133 const NcclCollectivePermuteConfig::SourceTargetMapEntry source_target = in RunNcclCollective()
142 NcclCollectivePermuteConfig::SourceTargetMapEntry source_target, in RunCollectivePermute()
H A Djitrt_custom_calls.cc2046 const NcclCollectivePermuteConfig::SourceTargetMapEntry source_target = in operator ()()