Searched refs:CompactShapeFunction (Results 1 – 2 of 2) sorted by relevance
44 using CompactShapeFunction = std::function<StatusOr<Shape>(const Shape&)>; variable88 CompactShapeFunction compact_shape_function = nullptr,192 const CompactShapeFunction compact_shape_function_;
500 const HloRematerialization::CompactShapeFunction& compact_shape_function,775 const HloRematerialization::CompactShapeFunction& compact_shape_function_;795 const HloRematerialization::CompactShapeFunction& compact_shape_function, in MemoryUsageTracker()