Searched refs:size_function_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization.h | 62 : size_function_(size_function), in HloRematerialization()
102 const ShapeSizeFunction size_function_; variable
|
| D | hlo_memory_scheduler.cc | 110 size_function_(size_function), in ListScheduler()
184 entry.bytes_defined += size_function_(*buffer); in MakeReadyListEntry()
227 freed_bytes += size_function_(*buffer); in BytesFreedIfScheduled()
372 const LogicalBuffer::SizeFunction& size_function_; member in xla::__anone86b2dc90111::ListScheduler
612 : size_function_(size_function), algorithm_(algorithm) {} in HloMemoryScheduler()
616 ScheduleModule(module, size_function_, algorithm_)); in Run()
|
| D | hlo_memory_scheduler.h | 107 LogicalBuffer::SizeFunction size_function_;
|
| D | hlo_rematerialization.cc | 938 MemoryUsageTracker tracker(computation, size_function_, *points_to_analysis_, in ComputePeakMemory()
981 MemoryUsageTracker memory_tracker(computation, size_function_, in RematerializeComputation()
1226 module_output_size += size_function_(subshape); in Run()
|