Searched refs:size_fn (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_memory_scheduler_test.cc | 122 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
127 ScheduleModule(module.get(), size_fn, ListMemoryScheduler)); in TEST_F()
284 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
289 ScheduleModule(module.get(), size_fn, ListMemoryScheduler)); in TEST_F()
304 *points_to_analysis, size_fn) in TEST_F()
310 *points_to_analysis, size_fn, &memory_by_computation) in TEST_F()
|
| D | buffer_value.cc | 54 LogicalBufferProto BufferValue::ToProto(const SizeFunction& size_fn) const { in ToProto()
57 proto.set_size(size_fn(*this)); in ToProto()
|
| D | heap_simulator.h | 119 const BufferValue::SizeFunction& size_fn,
131 const BufferValue::SizeFunction& size_fn,
141 const BufferValue::SizeFunction& size_fn,
|
| D | heap_simulator.cc | 77 const BufferValue::SizeFunction& size_fn, const Options& options) { in Run() argument
78 HeapSimulator heap(std::move(algorithm), size_fn, options, &schedule); in Run()
92 const BufferValue::SizeFunction& size_fn, const Options& options, in Run() argument
95 HeapSimulator heap(std::move(algorithm), size_fn, options, in Run()
351 const BufferValue::SizeFunction& size_fn, const Options& options, in HeapSimulator() argument
357 size_fn_(size_fn), in HeapSimulator()
|
| D | buffer_value.h | 150 LogicalBufferProto ToProto(const SizeFunction& size_fn) const;
|
| D | hlo_module_test.cc | 298 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
301 HloMemoryScheduler scheduler(size_fn); in TEST_F()
|
| D | heap_simulator_test.cc | 85 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
98 HeapSimulator::MinimumMemoryForModule(schedule, size_fn).ValueOrDie()); in TEST_F()
203 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
216 *points_to_analysis, size_fn, &memory_by_computation) in TEST_F()
308 auto size_fn = [&reverse_position](const BufferValue& buffer) { in RunWholeModule() local
314 *points_to_analysis_, size_fn) in RunWholeModule()
|
| /external/squashfs-tools/squashfs-tools/ |
| D | action.c | 3213 { "size", 1, size_fn, parse_number_arg, 1, 0},
|