HomeSort by relevance Sort by last modified time
    Searched defs:size_function_ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_rematerialization.h 77 size_function_(size_function) {}
114 const ShapeSizeFunction size_function_; member in class:xla::HloRematerialization
hlo_scheduling.cc 98 size_function_(size_function) {
170 entry.bytes_defined += size_function_(*buffer);
193 freed_bytes += size_function_(*buffer);
316 const LogicalBuffer::SizeFunction& size_function_; member in class:xla::__anon38971::ListScheduler

Completed in 131 milliseconds