Home
last modified time | relevance | path

Searched defs:size_function_ (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_rematerialization.h91 : size_function_(size_function), in size_function_() function
167 const ShapeSizeFunction size_function_; variable
H A Dhlo_memory_scheduler.cc384 const BufferValue::SizeFunction& size_function_; member in xla::__anon44666b960111::ListScheduler
H A Dhlo_rematerialization.cc771 const HloRematerialization::ShapeSizeFunction& size_function_; member in xla::__anon2b46f4430111::MemoryUsageTracker