tf_1.8_xla_doc
Namespaces | Functions
hlo_scheduling.cc File Reference
#include "tensorflow/compiler/xla/service/hlo_scheduling.h"
#include <map>
#include <utility>
#include <vector>
#include "tensorflow/compiler/xla/service/heap_simulator.h"
#include "tensorflow/compiler/xla/service/hlo_computation.h"
#include "tensorflow/compiler/xla/service/tuple_points_to_analysis.h"
#include "tensorflow/compiler/xla/shape_util.h"
#include "tensorflow/compiler/xla/status_macros.h"
#include "tensorflow/compiler/xla/statusor.h"
#include "tensorflow/compiler/xla/types.h"
#include "tensorflow/compiler/xla/util.h"
#include "tensorflow/core/lib/core/errors.h"
#include "tensorflow/core/lib/strings/str_util.h"
#include "tensorflow/core/lib/strings/stringprintf.h"
#include "tensorflow/core/platform/logging.h"
Include dependency graph for hlo_scheduling.cc:

Namespaces

 xla
 namespace for xla
 

Functions

StatusOr< std::vector< const HloInstruction * > > xla::anonymous_namespace{hlo_scheduling.cc}::CreateMemoryMinimizingSequence (const HloComputation &computation, const TuplePointsToAnalysis &points_to_analysis, const LogicalBuffer::SizeFunction &size_function, const MemorySchedulerAlgorithm &algorithm)
 
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)