| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_memory_scheduler_test.cc | 141 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
147 ScheduleModule(module.get(), size_fn, in TEST_F()
171 EXPECT_EQ(PeakMemoryUseOfEntryComputation(module.get(), size_fn), in TEST_F()
194 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
199 ScheduleModule(module.get(), size_fn, in TEST_F()
|
| D | buffer_value.cc | 48 LogicalBufferProto BufferValue::ToProto(const SizeFunction& size_fn) const { in ToProto()
51 proto.set_size(size_fn(*this)); in ToProto()
|
| D | heap_simulator.h | 153 const BufferValue::SizeFunction& size_fn,
165 const BufferValue::SizeFunction& size_fn,
177 const BufferValue::SizeFunction& size_fn, const HloSchedule* schedule,
185 const BufferValue::SizeFunction& size_fn,
|
| D | heap_simulator.cc | 103 const BufferValue::SizeFunction& size_fn, const Options& options) { in Run() argument
104 HeapSimulator heap(std::move(algorithm), size_fn, options, &schedule); in Run()
123 const BufferValue::SizeFunction& size_fn, const Options& options, in Run() argument
126 HeapSimulator heap(std::move(algorithm), size_fn, options, in Run()
144 const BufferValue::SizeFunction& size_fn, const HloSchedule* schedule, in Run() argument
146 HeapSimulator heap(std::move(algorithm), size_fn, options, in Run()
339 const BufferValue::SizeFunction& size_fn, const Options& options, in HeapSimulator() argument
346 size_fn_(size_fn), in HeapSimulator()
|
| D | buffer_value.h | 153 LogicalBufferProto ToProto(const SizeFunction& size_fn) const;
|
| D | hlo_module_test.cc | 422 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
425 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
217 *alias_analysis, size_fn, &memory_by_computation) in TEST_F()
321 auto size_fn = [&reverse_position](const BufferValue& buffer) { in RunWholeModule() local
326 *alias_analysis_, size_fn) in RunWholeModule()
|
| D | memory_space_assignment.cc | 114 options.size_fn(value) <= options.max_size_in_bytes && in IsCrossProgramPrefetchCandidate()
146 interval.size = options.size_fn(*value); in FindCrossProgramPrefetchCandidate()
1457 reserved_in_bytes_ += options_.size_fn(*interval.buffer); in Finish()
3413 options_.size_fn(dataflow_analysis->GetUniqueValueAt(instruction)); in CalculateAsyncCopyStats()
3508 options_.size_fn, in FindAllocationSequence()
|
| D | memory_space_assignment.h | 1031 BufferValue::SizeFunction size_fn; member
|
| D | memory_space_assignment_test.cc | 111 auto size_fn = [](const BufferValue& buffer) { in AssignMemorySpace() local
155 options.size_fn = size_fn; in AssignMemorySpace()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_op.h | 301 Status DoLazyPrepare(std::function<int64(void)> size_fn) override { in DoLazyPrepare() argument
302 return DoPrepare(size_fn()); in DoLazyPrepare()
|
| /external/squashfs-tools/squashfs-tools/ |
| D | action.c | 3213 { "size", 1, size_fn, parse_number_arg, 1, 0},
|