Home
last modified time | relevance | path

Searched refs:MinimumMemoryForComputation (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_memory_scheduler.cc519 *peak_memory, HeapSimulator::MinimumMemoryForComputation( in DFSMemoryScheduler()
574 *peak_memory, HeapSimulator::MinimumMemoryForComputation( in ListMemoryScheduler()
595 *peak_memory, HeapSimulator::MinimumMemoryForComputation( in PostOrderMemoryScheduler()
H A Dheap_simulator.h127 static StatusOr<int64_t> MinimumMemoryForComputation(
134 static StatusOr<int64_t> MinimumMemoryForComputation(
H A Dheap_simulator.cc72 StatusOr<int64_t> HeapSimulator::MinimumMemoryForComputation( in MinimumMemoryForComputation() function in xla::HeapSimulator
86 StatusOr<int64_t> HeapSimulator::MinimumMemoryForComputation( in MinimumMemoryForComputation() function in xla::HeapSimulator
H A Dheap_simulator_test.cc215 EXPECT_EQ(64, HeapSimulator::MinimumMemoryForComputation( in TEST_F()