Searched defs:size_function_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ | ||
| D | hlo_rematerialization.h | 102 const ShapeSizeFunction size_function_; variable |
| D | hlo_memory_scheduler.cc | 372 const LogicalBuffer::SizeFunction& size_function_; member in xla::__anone86b2dc90111::ListScheduler |