Searched refs:size_function_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization.h | 91 : size_function_(size_function), in size_function_() function
167 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::__anon454ff6160111::ListScheduler
761 : size_function_(size_function), algorithm_(algorithm) {} in HloMemoryScheduler()
768 ScheduleModule(module, size_function_, algorithm_, execution_threads)); in Run()
|
| D | hlo_rematerialization.cc | 758 buffer_id, defining_instruction, size_function_(shape), shape, live_out, in NewBuffer()
771 const HloRematerialization::ShapeSizeFunction& size_function_; member in xla::__anon4961cec30111::MemoryUsageTracker
801 size_function_(size_function), in MemoryUsageTracker()
957 int64_t compact_shape_size = size_function_(compact_shape); in MemoryReducedIfCompressed()
1026 memory_usage_ -= size_function_(original_item->instruction->shape()); in AddCompressInstructions()
1028 memory_usage_ += size_function_(compressed_item->instruction->shape()); in AddCompressInstructions()
1440 const int64_t size = size_function_(item->instruction->shape()); in PickRematerializationCandidates()
1441 const int64_t reduced_size = size_function_(compact_shape); in PickRematerializationCandidates()
1847 MemoryUsageTracker tracker(computation, size_function_, in ComputePeakMemory()
1905 computation, size_function_, compact_shape_function_, in RematerializeComputation()
[all …]
|
| D | hlo_memory_scheduler.h | 158 LogicalBuffer::SizeFunction size_function_;
|