Searched refs:min_remat_size (Results 1 – 3 of 3) sorted by relevance
89 int64 min_remat_size = 0)99 min_remat_size_(min_remat_size) {} in size_function_()120 int64 min_remat_size);
45 int64 min_remat_size = 0) { in RunHloRematerialization() argument57 min_remat_size); in RunHloRematerialization()602 int64 min_remat_size = 0) { in RunHloRematerialization() argument610 min_remat_size); in RunHloRematerialization()
1832 int64 memory_limit_bytes, int64 min_remat_size) { in RematerializeComputation() argument1845 return memory_tracker.AllocatedSize(item) >= min_remat_size; in RematerializeComputation()1957 min_remat_size)); in RematerializeComputation()