Home
last modified time | relevance | path

Searched defs:size_fn (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler_test.cc141 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
194 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
Dheap_simulator.cc100 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()
Dheap_simulator_test.cc86 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
204 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
321 auto size_fn = [&reverse_position](const BufferValue& buffer) { in RunWholeModule() local
Dhlo_module_test.cc296 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
Dmemory_space_assignment.h427 BufferValue::SizeFunction size_fn; member
Dmemory_space_assignment_test.cc90 auto size_fn = [](const BufferValue& buffer) { in AssignMemorySpace() local
/external/tensorflow/tensorflow/core/kernels/
Dlookup_table_op.h230 Status DoLazyPrepare(std::function<int64(void)> size_fn) override { in DoLazyPrepare()