Home
last modified time | relevance | path

Searched refs:HloRematerialization (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.h38 class HloRematerialization : public HloModulePass {
60 HloRematerialization(const ShapeSizeFunction& size_function, in HloRematerialization() function
65 ~HloRematerialization() {} in ~HloRematerialization()
Dhlo_rematerialization.cc329 const HloRematerialization::ShapeSizeFunction& size_function,
416 const HloRematerialization::ShapeSizeFunction& size_function, in CreateBufferFromLogicalBuffer()
510 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker()
934 StatusOr<int64> HloRematerialization::ComputePeakMemory( in ComputePeakMemory()
956 StatusOr<int64> HloRematerialization::CalledComputationsMemoryUsage( in CalledComputationsMemoryUsage()
971 StatusOr<bool> HloRematerialization::RematerializeComputation( in RematerializeComputation()
1203 StatusOr<bool> HloRematerialization::Run(HloModule* module) { in Run()
Dhlo_rematerialization_test.cc151 HloRematerialization remat(ByteSizeOf, memory_limit_bytes, in RunHloRematerialization()