Searched refs:compact_shape_function_ (Results 1 – 2 of 2) sorted by relevance
97 compact_shape_function_(compact_shape_function == nullptr in size_function_()192 const CompactShapeFunction compact_shape_function_; variable
775 const HloRematerialization::CompactShapeFunction& compact_shape_function_; member in xla::__anon2b46f4430111::MemoryUsageTracker802 compact_shape_function_(compact_shape_function), in MemoryUsageTracker()1258 TF_ASSIGN_OR_RETURN(Shape min_shape, compact_shape_function_(original_shape)); in GetCompactShape()1848 compact_shape_function_, *points_to_analysis_, in ComputePeakMemory()1905 computation, size_function_, compact_shape_function_, in RematerializeComputation()