tf_1.8_xla_doc
|
#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"
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) |