| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_value.h | 86 class BufferValue {
95 using SizeFunction = std::function<int64(const BufferValue&)>;
96 using AlignmentFunction = std::function<int64(BufferValue::Color)>;
98 virtual ~BufferValue();
144 bool operator<(const BufferValue& other) const { return id_ < other.id_; }
146 bool operator==(const BufferValue& other) const { return id_ == other.id_; }
147 bool operator!=(const BufferValue& other) const { return id_ != other.id_; }
162 BufferValue(HloInstruction* instruction, const ShapeIndex& index, Id id);
178 std::ostream& operator<<(std::ostream& out, const BufferValue& buffer);
|
| D | buffer_value.cc | 28 BufferValue::BufferValue(HloInstruction* instruction, const ShapeIndex& index, in BufferValue() function in xla::BufferValue
36 BufferValue::~BufferValue() {} in ~BufferValue()
38 std::ostream& operator<<(std::ostream& out, const BufferValue& buffer) { in operator <<()
43 /*static*/ LogicalBufferProto::Location BufferValue::ToLocationProto( in ToLocationProto()
54 LogicalBufferProto BufferValue::ToProto(const SizeFunction& size_fn) const { in ToProto()
|
| D | hlo_memory_scheduler.cc | 83 const BufferValue::SizeFunction& size_function, in Run()
108 const BufferValue::SizeFunction& size_function, in ListScheduler()
384 const BufferValue::SizeFunction& size_function_;
405 const BufferValue::SizeFunction& size_function) { in SumLogicalBufferSizes()
417 const BufferValue::SizeFunction& size_function, in ScheduleComputationHelper()
439 const BufferValue::SizeFunction& size_function, in DFSMemoryScheduler()
546 const BufferValue::SizeFunction& size_function, in ListMemoryScheduler()
566 const BufferValue::SizeFunction& size_function, in PostOrderMemoryScheduler()
584 const BufferValue::SizeFunction& size_function, in DefaultMemoryScheduler()
642 const BufferValue::SizeFunction& size_function, int64* peak_memory) { in DefaultModuleScheduler()
[all …]
|
| D | hlo_schedule_test.cc | 58 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
92 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
141 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
185 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
246 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
315 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
|
| D | buffer_value_containers.h | 30 tensorflow::gtl::CompactPointerSet<const BufferValue*>;
41 using BufferValueFlatSet = absl::flat_hash_set<const BufferValue*>;
|
| D | hlo_buffer.h | 98 BufferValue::Color color() const { in color()
100 BufferValue::Color result = values()[0]->color(); in color()
|
| D | hlo_memory_scheduler_test.cc | 90 HloMemoryScheduler scheduler([](const BufferValue& buffer) { in TEST_F()
141 auto size_fn = [](const BufferValue& buffer) { in TEST_F()
194 auto size_fn = [](const BufferValue& buffer) { in TEST_F()
247 [](const BufferValue& buffer) { in TEST_F()
298 [](const BufferValue& buffer) { 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,
214 const BufferValue::SizeFunction size_fn_;
|
| D | buffer_assignment.h | 499 BufferValue::SizeFunction buffer_size, in BufferAssignment()
570 BufferValue::SizeFunction buffer_size_;
594 value->set_color(BufferValue::Color( in DefaultColorer()
597 value->set_color(BufferValue::Color(0)); in DefaultColorer()
614 BufferValue::SizeFunction buffer_size,
635 BufferValue::SizeFunction buffer_size,
|
| D | logical_buffer.cc | 28 : BufferValue(instruction, index, id), in LogicalBuffer()
|
| D | logical_buffer.h | 35 class LogicalBuffer : public BufferValue {
|
| D | compiler.h | 295 std::function<int64(const BufferValue&)> BufferSizeBytesFunction() { in BufferSizeBytesFunction()
297 return [shape_size](const BufferValue& buffer) { in BufferSizeBytesFunction()
|
| D | buffer_assignment.cc | 614 flat_hash_map<BufferValue::Color, BufferAllocation*> combined_allocation_map; in CombineTempAllocations()
629 BufferValue::Color color = temp_allocation.color(); in CombineTempAllocations()
876 for (BufferValue::Id id = 0; id < dataflow.values().size(); id++) { in ToProto()
892 BufferValue::ToLocationProto(*alias->instruction(), alias->index()); in ToProto()
911 BufferValue::SizeFunction buffer_size, in Run()
1508 absl::flat_hash_map<BufferValue::Id, const HloValue*> id_to_value; in ComputePeakMemoryLogicalBuffers()
1595 BufferValue::Color color) { in AssignBuffersFromHeapSimulator()
1627 BufferValue::SizeFunction buffer_size, in CreateAssignment()
|
| D | heap_simulator.cc | 100 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()
|
| D | hlo_value.h | 100 class HloValue : public BufferValue {
|
| D | hlo_value.cc | 73 : BufferValue(instruction, index, id), is_phi_(is_phi) { in HloValue()
|
| D | heap_simulator_test.cc | 86 auto size_fn = [](const BufferValue& buffer) { in TEST_F()
204 auto size_fn = [](const BufferValue& buffer) { in TEST_F()
321 auto size_fn = [&reverse_position](const BufferValue& buffer) { in RunWholeModule()
385 auto zero_size = [](const BufferValue& buffer) { return 0; }; in Init()
|
| D | hlo_module_test.cc | 296 auto size_fn = [](const BufferValue& buffer) { in TEST_F()
|
| D | hlo_rematerialization_test.cc | 48 [](const BufferValue& buffer) { return ByteSizeOf(buffer.shape()); }, in RunHloRematerialization()
|
| D | memory_space_assignment.h | 427 BufferValue::SizeFunction size_fn;
|
| D | buffer_assignment_test.cc | 590 value.set_color(BufferValue::Color(color++)); in TEST_F()
2049 static int64 ByteSizeOf(const BufferValue& buffer) { in ByteSizeOf()
2349 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | buffer_assignment_util.cc | 26 const BufferValue* buffer = buffer_offset_pair.first; in InstrForConstantBufferAllocation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_schedule.cc | 205 entry_computation, [pointer_size](const BufferValue& buffer) { in Build()
|
| D | gpu_compiler.cc | 528 [this](const BufferValue& buffer_value) -> int64 { in RunHloPassesAndBufferAssignement()
571 [pointer_size](const BufferValue& buffer_value) -> int64 { in CompileModuleToLlvmIrImpl()
|
| /external/deqp/external/vulkancts/modules/vulkan/binding_model/ |
| D | vktBindingDescriptorCopyTests.cpp | 605 union BufferValue in init() union
|