Searched defs:size_fn (Results 1 – 7 of 7) sorted by relevance
141 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local194 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
100 const BufferValue::SizeFunction& size_fn, const Options& options) { in Run()120 const BufferValue::SizeFunction& size_fn, const Options& options, in Run()141 const BufferValue::SizeFunction& size_fn, const HloSchedule* schedule, in Run()336 const BufferValue::SizeFunction& size_fn, const Options& options, in HeapSimulator()
86 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local204 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local321 auto size_fn = [&reverse_position](const BufferValue& buffer) { in RunWholeModule() local
296 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
427 BufferValue::SizeFunction size_fn; member
90 auto size_fn = [](const BufferValue& buffer) { in AssignMemorySpace() local
230 Status DoLazyPrepare(std::function<int64(void)> size_fn) override { in DoLazyPrepare()