Searched defs:size_function_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ | ||
| D | hlo_rematerialization.h | 88 : size_function_(size_function), in size_function_() function |
| D | hlo_memory_scheduler.cc | 372 const BufferValue::SizeFunction& size_function_; member in xla::__anonc7f641cb0111::ListScheduler |
| D | hlo_rematerialization.cc | 599 const HloRematerialization::ShapeSizeFunction& size_function_; member in xla::__anon20d191180111::MemoryUsageTracker |