Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.h34 struct FusionInfoCache { struct
84 FusionInfoCache* cache = nullptr);
Dmulti_output_fusion.cc55 FusionInfoCache* fusion_info_cache) { in LegalToFuse()
113 FusionInfoCache* fusion_info_cache) { in GetProducerConsumerMultiOutputFusionCandidates()
203 FusionInfoCache* fusion_info_cache) { in FuseSiblings()
272 FusionInfoCache fusion_info_cache; in DoMultiOutputFusion()
Dmulti_output_fusion.h103 bool FuseSiblings(HloInstruction* parent, FusionInfoCache* fusion_info_cache);
Dgpu_fusible.cc326 FusionInfoCache* cache = nullptr) { in SharedMemoryUsage()
367 FusionInfoCache* cache) { in NumUnnestedReductions()
412 FusionInfoCache* cache /*=nullptr*/) { in FusionFitsInBudget()