HomeSort by relevance Sort by last modified time
    Searched refs:FusionMerger (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
fusion_merger.h 35 class FusionMerger : public HloPassInterface {
fusion_merger_test.cc 127 const int64 flops_needed = FusionMerger::GetThresholdFlopsToBytesRatio() *
289 EXPECT_TRUE(FusionMerger().Run(module_.get()).ValueOrDie());
352 EXPECT_FALSE(FusionMerger().Run(module_.get()).ValueOrDie());
403 EXPECT_FALSE(FusionMerger().Run(module_.get()).ValueOrDie());
459 EXPECT_TRUE(FusionMerger().Run(module_.get()).ValueOrDie());
fusion_merger.cc 255 FusionMerger::GetThresholdFlopsToBytesRatio()) {
293 StatusOr<bool> FusionMerger::Run(HloModule* module) {
295 VLOG(2) << "FusionMerger for module: " << module->name();
gpu_compiler.cc 249 fusion.AddPass<FusionMerger>();

Completed in 119 milliseconds