| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_memory_scheduler.cc | 83 const BufferValue::SizeFunction& size_function, in Run() argument
86 ListScheduler scheduler(computation, points_to_analysis, size_function, in Run()
108 const BufferValue::SizeFunction& size_function, in ListScheduler() argument
113 size_function_(size_function), in ListScheduler()
405 const BufferValue::SizeFunction& size_function) { in SumLogicalBufferSizes() argument
408 size += size_function(*buffer); in SumLogicalBufferSizes()
417 const BufferValue::SizeFunction& size_function, in ScheduleComputationHelper() argument
426 size_function, memory_by_computation, postprocessor, in ScheduleComputationHelper()
430 size_function, memory_by_computation, in ScheduleComputationHelper()
440 const BufferValue::SizeFunction& size_function, in DFSMemoryScheduler() argument
[all …]
|
| D | hlo_memory_scheduler.h | 75 const LogicalBuffer::SizeFunction& size_function,
85 const LogicalBuffer::SizeFunction& size_function,
95 const LogicalBuffer::SizeFunction& size_function,
108 const LogicalBuffer::SizeFunction& size_function,
116 const LogicalBuffer::SizeFunction& size_function,
125 const HloModule* module, const LogicalBuffer::SizeFunction& size_function,
134 const LogicalBuffer::SizeFunction& size_function,
145 HloMemoryScheduler(const LogicalBuffer::SizeFunction& size_function,
|
| D | hlo_rematerialization.h | 85 const ShapeSizeFunction& size_function, int64_t memory_limit_bytes,
91 : size_function_(size_function), in size_function_() argument
|
| D | heap_simulator.h | 123 const LogicalBuffer::SizeFunction& size_function);
130 const LogicalBuffer::SizeFunction& size_function,
137 const LogicalBuffer::SizeFunction& size_function,
|
| D | heap_simulator.cc | 50 const LogicalBuffer::SizeFunction& size_function) { in MinimumMemoryForModule() argument
67 *module, schedule, *alias_analysis, size_function)); in MinimumMemoryForModule()
75 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation() argument
81 computation, sequence, alias_analysis, size_function, in MinimumMemoryForComputation()
89 const LogicalBuffer::SizeFunction& size_function, in MinimumMemoryForComputation() argument
94 computation, sequence, alias_analysis, size_function, in MinimumMemoryForComputation()
|
| D | hlo_memory_scheduler_test.cc | 42 HloModule* module, LogicalBuffer::SizeFunction size_function) { in PeakMemoryUseOfEntryComputation() argument
55 *computation, sequence, *alias_analysis, size_function) in PeakMemoryUseOfEntryComputation()
|
| D | hlo_rematerialization.cc | 499 const HloRematerialization::ShapeSizeFunction& size_function,
794 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker() argument
801 size_function_(size_function), in MemoryUsageTracker()
|
| /external/openthread/third_party/mbedtls/repo/docs/proposed/ |
| D | psa-driver-interface.md | 609 * `"size_function"` (string, optional): the name of a function that returns the number of bytes tha…
610 …nal): If specified, this overrides all other methods (including the `"size_function"` entry point)…
616 …xt and chooses to use it, and the driver specification includes the `"size_function"` property, th…
618 size_function(key_type, key_bits)
620 where `size_function` is the function named in the `"size_function"` property, `key_type` is the ke…
622 size_t size_function(psa_key_type_t key_type, size_t key_bits);
627 … not to use it, or if the driver specification does not include the `"size_function"` property, th…
|
| /external/mbedtls/docs/proposed/ |
| D | psa-driver-interface.md | 616 * `"size_function"` (string, optional): the name of a function that returns the number of bytes tha…
617 …nal): If specified, this overrides all other methods (including the `"size_function"` entry point)…
623 …xt and chooses to use it, and the driver specification includes the `"size_function"` property, th…
625 size_function(key_type, key_bits)
627 where `size_function` is the function named in the `"size_function"` property, `key_type` is the ke…
629 size_t size_function(psa_key_type_t key_type, size_t key_bits);
634 … not to use it, or if the driver specification does not include the `"size_function"` property, th…
|