HomeSort by relevance Sort by last modified time
    Searched refs:size_function_ (Results 1 - 3 of 3) 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
hlo_rematerialization.cc     [all...]

Completed in 966 milliseconds