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