Searched refs:HloReachabilityMap (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_reachability.cc | 22 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::SetReachable(const HloInstruction* a, in SetReachable()
68 bool HloReachabilityMap::IsReachable(const HloInstruction* a, in IsReachable()
73 bool HloReachabilityMap::IsConnected(const HloInstruction* a, in IsConnected()
78 std::unique_ptr<HloReachabilityMap> HloReachabilityMap::Build( in Build()
81 auto result = absl::make_unique<HloReachabilityMap>(all); in Build()
140 void HloReachabilityMap::UpdateReachabilityThroughInstruction( in UpdateReachabilityThroughInstruction()
|
| D | hlo_ordering.h | 105 HloReachabilityMap& reachability_map(const HloComputation* computation) { in reachability_map()
108 const HloReachabilityMap& reachability_map( in reachability_map()
125 std::unique_ptr<HloReachabilityMap>>
|
| D | hlo_reachability.h | 43 class HloReachabilityMap {
47 explicit HloReachabilityMap(
56 static std::unique_ptr<HloReachabilityMap> Build(
|
| D | hlo_module_group_util.h | 102 StatusOr<std::unique_ptr<HloReachabilityMap>> ComputeReachability(
108 HloInstruction* instruction, HloReachabilityMap* reachability_map);
|
| D | multi_output_fusion.h | 89 HloReachabilityMap* reachability() const { return reachability_.get(); } in reachability()
134 std::unique_ptr<HloReachabilityMap> reachability_;
|
| D | hlo_reachability_test.cc | 54 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()
|
| D | hlo_module_group_util.cc | 347 StatusOr<std::unique_ptr<HloReachabilityMap>>
363 auto reachability = absl::make_unique<HloReachabilityMap>(post_order); in ComputeReachability()
371 HloInstruction* instruction, HloReachabilityMap* reachability_map) { in UpdateReachabilityThroughInstruction()
|
| D | instruction_fusion.h | 122 std::unique_ptr<HloReachabilityMap> reachability_;
|
| D | multi_output_fusion.cc | 261 reachability_ = HloReachabilityMap::Build(computation_); in RecomputeReachability()
|
| D | hlo_ordering.cc | 341 predecessors_.emplace(computation, HloReachabilityMap::Build(computation)); in DependencyHloOrdering()
|
| D | instruction_fusion.cc | 457 reachability_ = HloReachabilityMap::Build(computation_); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_assignment.cc | 51 const HloReachabilityMap& reachability) { in CanRunConcurrently()
67 const HloReachabilityMap& reachability, in ComputeStreamToAssign()
125 std::unique_ptr<HloReachabilityMap> reachability = in AssignStreams()
126 HloReachabilityMap::Build(&computation); in AssignStreams()
|
| D | gpu_hlo_schedule.cc | 79 auto predecessor_map = absl::make_unique<HloReachabilityMap>( in GpuHloOrdering()
128 HloReachabilityMap::Build(computation)); in GpuHloOrdering()
|