| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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 | 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 | heap_simulator.h | 147 const BufferValue::SizeFunction& size_fn,
159 const BufferValue::SizeFunction& size_fn,
171 const BufferValue::SizeFunction& size_fn, const HloSchedule* schedule,
179 const BufferValue::SizeFunction& size_fn,
|
| D | heap_simulator.cc | 100 const BufferValue::SizeFunction& size_fn, const Options& options) { in Run() argument
101 HeapSimulator heap(std::move(algorithm), size_fn, options, &schedule); in Run()
120 const BufferValue::SizeFunction& size_fn, const Options& options, in Run() argument
123 HeapSimulator heap(std::move(algorithm), size_fn, options, in Run()
141 const BufferValue::SizeFunction& size_fn, const HloSchedule* schedule, in Run() argument
143 HeapSimulator heap(std::move(algorithm), size_fn, options, in Run()
336 const BufferValue::SizeFunction& size_fn, const Options& options, in HeapSimulator() argument
343 size_fn_(size_fn), in HeapSimulator()
|
| D | buffer_value.h | 152 LogicalBufferProto ToProto(const SizeFunction& size_fn) const;
|
| D | hlo_module_test.cc | 296 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
299 HloMemoryScheduler scheduler(size_fn); in TEST_F()
|
| D | heap_simulator_test.cc | 86 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
99 HeapSimulator::MinimumMemoryForModule(schedule, size_fn).ValueOrDie()); in TEST_F()
204 auto size_fn = [](const BufferValue& buffer) { in TEST_F() local
218 *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 | 73 options.size_fn(value) <= options.max_size_in_bytes && in IsCrossProgramPrefetchCandidate()
103 interval.size = options.size_fn(*value); in FindCrossProgramPrefetchCandidate()
1121 reserved_in_bytes_ += options_.size_fn(*interval.buffer); in Finish()
2611 options_.size_fn(dataflow_analysis->GetUniqueValueAt(instruction)); in CalculateAsyncCopyStats()
2699 options_.size_fn, in FindAllocationSequence()
|
| D | memory_space_assignment.h | 427 BufferValue::SizeFunction size_fn; member
|
| D | memory_space_assignment_test.cc | 90 auto size_fn = [](const BufferValue& buffer) { in AssignMemorySpace() local
134 options.size_fn = size_fn; in AssignMemorySpace()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_op.h | 230 Status DoLazyPrepare(std::function<int64(void)> size_fn) override { in DoLazyPrepare() argument
231 return DoPrepare(size_fn()); in DoLazyPrepare()
|
| /external/squashfs-tools/squashfs-tools/ |
| D | action.c | 3213 { "size", 1, size_fn, parse_number_arg, 1, 0},
|