Searched refs:RematerializeComputation (Results 1 – 2 of 2) sorted by relevance
82 StatusOr<bool> RematerializeComputation(HloComputation* computation,
971 StatusOr<bool> HloRematerialization::RematerializeComputation( in RematerializeComputation() function in xla::HloRematerialization1154 RematerializeComputation(called_computation, schedule, in RematerializeComputation()1264 RematerializeComputation(module->entry_computation(), &module->schedule(), in Run()