Searched refs:ComputePeakMemory (Results 1 – 2 of 2) sorted by relevance
90 StatusOr<int64> ComputePeakMemory(const HloComputation* computation,
934 StatusOr<int64> HloRematerialization::ComputePeakMemory( in ComputePeakMemory() function in xla::HloRematerialization1243 ComputePeakMemory(node.computation(), module->schedule().sequence( in Run()