Home
last modified time | relevance | path

Searched refs:compact_shape_function (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_rematerialization.h88 CompactShapeFunction compact_shape_function = nullptr,
97 compact_shape_function_(compact_shape_function == nullptr in size_function_()
99 : std::move(compact_shape_function)), in size_function_()
H A Dhlo_rematerialization.cc500 const HloRematerialization::CompactShapeFunction& compact_shape_function,
795 const HloRematerialization::CompactShapeFunction& compact_shape_function, in MemoryUsageTracker() argument
802 compact_shape_function_(compact_shape_function), in MemoryUsageTracker()