Searched refs:compact_shape_function_ (Results 1 – 2 of 2) sorted by relevance
95 compact_shape_function_(compact_shape_function == nullptr in size_function_()158 const CompactShapeFunction compact_shape_function_; variable
764 const HloRematerialization::CompactShapeFunction& compact_shape_function_; member in xla::__anon95ba67740111::MemoryUsageTracker791 compact_shape_function_(compact_shape_function), in MemoryUsageTracker()1247 TF_ASSIGN_OR_RETURN(Shape min_shape, compact_shape_function_(original_shape)); in GetCompactShape()1797 compact_shape_function_, *points_to_analysis_, in ComputePeakMemory()1841 computation, size_function_, compact_shape_function_, in RematerializeComputation()