| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_memory_scheduler.h | 39 HloComputation*, const TuplePointsToAnalysis&,
47 const TuplePointsToAnalysis& points_to_analysis,
55 const TuplePointsToAnalysis& points_to_analysis,
63 const TuplePointsToAnalysis& points_to_analysis,
73 const TuplePointsToAnalysis& points_to_analysis,
|
| tuple_points_to_analysis.cc | 137 /* static */ StatusOr<std::unique_ptr<TuplePointsToAnalysis>>
138 TuplePointsToAnalysis::Run(const HloModule* module) {
140 std::unique_ptr<TuplePointsToAnalysis> analysis(new TuplePointsToAnalysis(
146 Status TuplePointsToAnalysis::Analyze() {
177 Status TuplePointsToAnalysis::PopulateDefinedBuffersAndAliases(const decltype(
198 Status TuplePointsToAnalysis::DefaultAction(HloInstruction* hlo_instruction) {
219 Status TuplePointsToAnalysis::HandleGetTupleElement(
251 Status TuplePointsToAnalysis::HandleCopy(HloInstruction* copy) {
264 Status TuplePointsToAnalysis::HandleBitcast(HloInstruction* bitcast)
[all...] |
| buffer_liveness.h | 61 const TuplePointsToAnalysis& points_to_analysis() const {
109 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
|
| hlo_memory_scheduler.cc | 79 const TuplePointsToAnalysis& points_to_analysis,
104 const TuplePointsToAnalysis& points_to_analysis,
371 const TuplePointsToAnalysis& points_to_analysis_;
392 const TuplePointsToAnalysis::BufferDefinitionVector& buffers,
403 const TuplePointsToAnalysis& points_to_analysis,
421 const TuplePointsToAnalysis& points_to_analysis,
492 const TuplePointsToAnalysis& points_to_analysis,
502 const TuplePointsToAnalysis& points_to_analysis,
511 const TuplePointsToAnalysis& points_to_analysis,
574 TF_ASSIGN_OR_RETURN(std::unique_ptr<TuplePointsToAnalysis> points_to_analysis
[all...] |
| heap_simulator.h | 101 const TuplePointsToAnalysis& points_to_analysis,
118 const TuplePointsToAnalysis& points_to_analysis,
130 const TuplePointsToAnalysis& points_to_analysis,
149 const TuplePointsToAnalysis& points_to_analysis);
228 // Since TuplePointsToAnalysis is being replaced with a module-aware alias
|
| hlo_rematerialization.h | 121 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
|
| tuple_points_to_analysis.h | 151 // TuplePointsToAnalysis, so disable copying.
183 class TuplePointsToAnalysis : public DfsHloVisitorWithDefault {
186 static StatusOr<std::unique_ptr<TuplePointsToAnalysis>> Run(
239 // matches the TuplePointsToAnalysis' LogicalBuffer with 'buffer.id'. Returns
277 explicit TuplePointsToAnalysis(
359 TF_DISALLOW_COPY_AND_ASSIGN(TuplePointsToAnalysis);
|
| hlo_module_group_metadata.h | 210 TuplePointsToAnalysis* points_to_analysis(HloModule* module) const {
285 absl::flat_hash_map<HloModule*, std::unique_ptr<TuplePointsToAnalysis>>
|
| layout_assignment.h | 132 LayoutConstraints(const TuplePointsToAnalysis& points_to_analysis,
138 const TuplePointsToAnalysis& points_to_analysis() const {
226 const TuplePointsToAnalysis& points_to_analysis_;
399 const TuplePointsToAnalysis& points_to_analysis,
|
| heap_simulator.cc | 41 TF_ASSIGN_OR_RETURN(std::unique_ptr<TuplePointsToAnalysis> points_to_analysis,
42 TuplePointsToAnalysis::Run(module));
59 const TuplePointsToAnalysis& points_to_analysis,
76 const TuplePointsToAnalysis& points_to_analysis,
91 const TuplePointsToAnalysis& points_to_analysis,
107 const TuplePointsToAnalysis& points_to_analysis) {
175 const TuplePointsToAnalysis::BufferDefinitionVector&
[all...] |
| buffer_liveness.cc | 47 TF_ASSIGN_OR_RETURN(points_to_analysis_, TuplePointsToAnalysis::Run(module_));
|
| hlo_rematerialization.cc | 291 const TuplePointsToAnalysis& points_to_analysis,
330 const TuplePointsToAnalysis& points_to_analysis,
376 // TuplePointsToAnalysis which owns all LogicalBuffers cannot be updated after
415 const TuplePointsToAnalysis& points_to_analysis,
511 const TuplePointsToAnalysis& points_to_analysis,
[all...] |
| hlo_memory_scheduler_test.cc | 298 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis =
299 TuplePointsToAnalysis::Run(module.get()).ValueOrDie();
|
| hlo_module_group_metadata.cc | 136 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis,
137 TuplePointsToAnalysis::Run(module));
|
| heap_simulator_test.cc | 209 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis =
210 TuplePointsToAnalysis::Run(module.get()).ValueOrDie();
267 TuplePointsToAnalysis::Run(module_.get()).ConsumeValueOrDie();
292 TuplePointsToAnalysis::Run(module_.get()).ConsumeValueOrDie();
350 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
[all...] |
| layout_assignment.cc | 104 const TuplePointsToAnalysis& points_to_analysis,
[all...] |
| buffer_assignment.h | 327 // TuplePointsToAnalysis. That is, an allocation marked `maybe_live_out_`
438 const TuplePointsToAnalysis& points_to_analysis() const {
|
| tuple_points_to_analysis_test.cc | 59 TuplePointsToAnalysis::Run(module_.get()).ConsumeValueOrDie();
120 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
[all...] |
| buffer_assignment.cc | 67 const TuplePointsToAnalysis& points_to_analysis,
696 // NOTE: TuplePointsToAnalysis state is serialized here in BufferAssigment,
699 const TuplePointsToAnalysis& points_to_analysis =
[all...] |
