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::__anon74ea0e200111::ListScheduler |
| D | hlo_rematerialization.cc | 760 const HloRematerialization::ShapeSizeFunction& size_function_; member in xla::__anon6c3ee80d0111::MemoryUsageTracker |