#include <vector>
#include "tensorflow/compiler/xla/service/hlo_instruction.h"
#include "tensorflow/compiler/xla/service/hlo_module.h"
#include "tensorflow/compiler/xla/service/hlo_ordering.h"
#include "tensorflow/compiler/xla/service/logical_buffer.h"
#include "tensorflow/compiler/xla/service/tuple_points_to_analysis.h"
#include "tensorflow/compiler/xla/statusor.h"
#include "tensorflow/compiler/xla/types.h"
Go to the source code of this file.
|
StatusOr< std::vector< const HloInstruction * > > | xla::DefaultMemoryScheduler (const HloComputation &computation, const TuplePointsToAnalysis &points_to_analysis, const LogicalBuffer::SizeFunction &size_function) |
|
StatusOr< SequentialHloOrdering::HloModuleSequence > | xla::CreateMemoryMinimizingSequence (const HloModule &module, const LogicalBuffer::SizeFunction &size_function, const MemorySchedulerAlgorithm &algorithm) |
|