Searched refs:size_function_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization.h | 90 : size_function_(size_function), in size_function_() function
138 const ShapeSizeFunction size_function_; variable
|
| D | hlo_memory_scheduler.cc | 113 size_function_(size_function), in ListScheduler()
188 entry.bytes_defined += size_function_(*buffer); in MakeReadyListEntry()
232 freed_bytes += size_function_(*buffer); in BytesFreedIfScheduled()
384 const BufferValue::SizeFunction& size_function_; member in xla::__anon85afbca70111::ListScheduler
731 : size_function_(size_function), algorithm_(algorithm) {} in HloMemoryScheduler()
735 ScheduleModule(module, size_function_, algorithm_)); in Run()
|
| D | hlo_rematerialization.cc | 747 buffer_id, defining_instruction, size_function_(shape), shape, live_out, in NewBuffer()
760 const HloRematerialization::ShapeSizeFunction& size_function_; member in xla::__anon95ba67740111::MemoryUsageTracker
790 size_function_(size_function), in MemoryUsageTracker()
946 int64 compact_shape_size = size_function_(compact_shape); in MemoryReducedIfCompressed()
1015 memory_usage_ -= size_function_(original_item->instruction->shape()); in AddCompressInstructions()
1017 memory_usage_ += size_function_(compressed_item->instruction->shape()); in AddCompressInstructions()
1796 MemoryUsageTracker tracker(computation, size_function_, in ComputePeakMemory()
1841 computation, size_function_, compact_shape_function_, in RematerializeComputation()
2028 module_output_size += size_function_(subshape); in Run()
|
| D | hlo_memory_scheduler.h | 143 LogicalBuffer::SizeFunction size_function_;
|