Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_reachability.cc22 HloReachabilityMap::HloReachabilityMap( in HloReachabilityMap() function in xla::HloReachabilityMap
33 bool HloReachabilityMap::SetReachabilityToUnion( in SetReachabilityToUnion()
42 void HloReachabilityMap::FastSetReachabilityToUnion( in FastSetReachabilityToUnion()
48 void HloReachabilityMap::SetReachabilityToUnionHelper( in SetReachabilityToUnionHelper()
63 void HloReachabilityMap::Replace(const HloInstruction* original, in Replace()
72 void HloReachabilityMap::SetReachable(Index a, Index b) { in SetReachable()
76 std::unique_ptr<HloReachabilityMap> HloReachabilityMap::BuildWithRestrictions( in BuildWithRestrictions()
82 auto result = absl::make_unique<HloReachabilityMap>(all); in BuildWithRestrictions()
93 std::unique_ptr<HloReachabilityMap> HloReachabilityMap::Build( in Build()
96 auto result = absl::make_unique<HloReachabilityMap>(all); in Build()
[all …]
Dhlo_reachability.h43 class HloReachabilityMap {
47 explicit HloReachabilityMap(
56 static std::unique_ptr<HloReachabilityMap> Build(
65 static std::unique_ptr<HloReachabilityMap> BuildWithRestrictions(
95 friend class HloReachabilityMap;
Dhlo_ordering.h126 HloReachabilityMap& reachability_map(const HloComputation* computation) { in reachability_map()
129 const HloReachabilityMap& reachability_map( in reachability_map()
146 std::unique_ptr<HloReachabilityMap>>
Dhlo_module_group_util.h102 StatusOr<std::unique_ptr<HloReachabilityMap>> ComputeReachability(
108 HloInstruction* instruction, HloReachabilityMap* reachability_map);
Dhlo_reachability_test.cc54 HloReachabilityMap reachability({a, b, c, d, e}); in TEST_F()
123 auto reachability = HloReachabilityMap::Build(computation); in TEST_F()
203 auto reachability = HloReachabilityMap::Build(computation); in TEST_F()
219 auto reachability = HloReachabilityMap::Build(computation); in TEST_F()
Dmulti_output_fusion.h95 HloReachabilityMap* reachability() const { return reachability_.get(); } in reachability()
203 std::unique_ptr<HloReachabilityMap> reachability_;
Dhlo_module_group_util.cc342 StatusOr<std::unique_ptr<HloReachabilityMap>>
358 auto reachability = absl::make_unique<HloReachabilityMap>(post_order); in ComputeReachability()
366 HloInstruction* instruction, HloReachabilityMap* reachability_map) { in UpdateReachabilityThroughInstruction()
Dinstruction_fusion.h136 std::unique_ptr<HloReachabilityMap> reachability_;
Dwhile_loop_analysis.cc201 std::unique_ptr<HloReachabilityMap> hrm = in GetAuxiliaryLoopInductionVars()
202 HloReachabilityMap::BuildWithRestrictions( in GetAuxiliaryLoopInductionVars()
Dall_reduce_combiner.cc289 auto reachability = HloReachabilityMap::Build(computation); in Run()
Dmulti_output_fusion.cc363 reachability_ = HloReachabilityMap::Build(computation_); in RecomputeReachability()
Dhlo_ordering.cc478 predecessors_.emplace(computation, HloReachabilityMap::Build(computation)); in DependencyHloOrdering()
Dinstruction_fusion.cc492 reachability_ = HloReachabilityMap::Build(computation_); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_assignment.cc52 const HloReachabilityMap& reachability) { in CanRunConcurrently()
68 const HloReachabilityMap& reachability, in ComputeStreamToAssign()
130 std::unique_ptr<HloReachabilityMap> reachability = in AssignStreams()
131 HloReachabilityMap::Build(&computation); in AssignStreams()
Dgpu_hlo_schedule.cc79 auto predecessor_map = absl::make_unique<HloReachabilityMap>( in GpuHloOrdering()
128 HloReachabilityMap::Build(computation)); in GpuHloOrdering()
Dmulti_output_fusion.h110 std::unique_ptr<HloReachabilityMap> reachability_;
Dmulti_output_fusion.cc107 const HloInstruction* producer, const HloReachabilityMap& reachability) { in GetProducerConsumerMultiOutputFusionCandidates()
191 reachability_ = HloReachabilityMap::Build(computation_); in RecomputeReachability()
Dir_emitter_unnested.cc5491 std::unique_ptr<HloReachabilityMap> reachability_map = in DivideOutputInstructionsIntoGroups()
5492 HloReachabilityMap::Build(fused_computation); in DivideOutputInstructionsIntoGroups()