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