| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | heap_simulator.h | 61 absl::flat_hash_map<const BufferValue*, Chunk> chunk_map;
119 const BufferValue::SizeFunction& size_fn,
131 const BufferValue::SizeFunction& size_fn,
141 const BufferValue::SizeFunction& size_fn,
151 bool IgnoreBuffer(const BufferValue* buffer) const;
152 void Alloc(const BufferValue* buffer, const HloInstruction* instruction);
153 void Free(const BufferValue* buffer, const HloInstruction* instruction);
154 void ShareBuffer(const BufferValue* buffer, const BufferValue* shared,
159 const BufferValue* buffer,
161 const BufferValue* share_with_canonical);
[all …]
|
| D | buffer_value.h | 87 class BufferValue {
96 using SizeFunction = std::function<int64(const BufferValue&)>;
97 using AlignmentFunction = std::function<int64(BufferValue::Color)>;
99 virtual ~BufferValue();
142 bool operator<(const BufferValue& other) const { return id_ < other.id_; }
144 bool operator==(const BufferValue& other) const { return id_ == other.id_; }
145 bool operator!=(const BufferValue& other) const { return id_ != other.id_; }
160 BufferValue(HloInstruction* instruction, const ShapeIndex& index, Id id);
176 std::ostream& operator<<(std::ostream& out, const BufferValue& buffer);
|
| D | heap_simulator.cc | 77 const BufferValue::SizeFunction& size_fn, const Options& options) { in Run()
92 const BufferValue::SizeFunction& size_fn, const Options& options, in Run()
120 flat_hash_map<const BufferValue*, flat_hash_set<const HloInstruction*>> in RunComputation()
122 flat_hash_map<const HloInstruction*, flat_hash_set<const BufferValue*>> in RunComputation()
126 const BufferValue* buffer) { in RunComputation()
146 for (const BufferValue* buffer : buffer_set) { in RunComputation()
154 for (const BufferValue* buffer : points_to.element({})) { in RunComputation()
159 for (const BufferValue* buffer : gte_points_to.CreateFlattenedSet()) { in RunComputation()
171 std::vector<const BufferValue*> dead_buffers_to_free; in RunComputation()
172 std::vector<const BufferValue*> operand_buffers_to_free; in RunComputation()
[all …]
|
| 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_schedule_test.cc | 59 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
93 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
142 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
186 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
247 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
316 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
|
| D | heap_simulator_test.cc | 85 auto size_fn = [](const BufferValue& buffer) { in TEST_F()
203 auto size_fn = [](const BufferValue& buffer) { in TEST_F()
225 using CallSequence = std::vector<std::pair<string, const BufferValue*>>;
233 void Alloc(const BufferValue* buffer, int64 size) override { in Alloc()
241 void Free(const BufferValue* buffer, int64 size) override { in Free()
272 auto zero_size = [](const BufferValue& buffer) { return 0; }; in HeapSimulatorTracker()
308 auto size_fn = [&reverse_position](const BufferValue& buffer) { in RunWholeModule()
321 const BufferValue* BufferAt(const HloInstruction* instruction, in BufferAt()
328 const BufferValue* buffer = BufferAt(instruction, index); in OffsetAt()
788 const BufferValue* buffer_a_;
[all …]
|
| D | buffer_value_containers.h | 30 tensorflow::gtl::CompactPointerSet<const BufferValue*>;
41 using BufferValueFlatSet = absl::flat_hash_set<const BufferValue*>;
|
| D | hlo_memory_scheduler_test.cc | 71 HloMemoryScheduler scheduler([](const BufferValue& buffer) { in TEST_F()
122 auto size_fn = [](const BufferValue& buffer) { in TEST_F()
180 [](const BufferValue& buffer) { in TEST_F()
231 [](const BufferValue& buffer) { in TEST_F()
284 auto size_fn = [](const BufferValue& buffer) { in TEST_F()
|
| D | compiler.h | 250 std::function<int64(const BufferValue&)> BufferSizeBytesFunction() { in BufferSizeBytesFunction()
252 return [shape_size](const BufferValue& buffer) { in BufferSizeBytesFunction()
|
| D | logical_buffer.cc | 28 : BufferValue(instruction, index, id), in LogicalBuffer()
|
| D | logical_buffer.h | 35 class LogicalBuffer : public BufferValue {
|
| D | hlo_value.h | 88 class HloValue : public BufferValue {
|
| D | hlo_value.cc | 72 : BufferValue(instruction, index, id), is_phi_(is_phi) { in HloValue()
|
| D | hlo_module_test.cc | 298 auto size_fn = [](const BufferValue& buffer) { in TEST_F()
|
| D | hlo_rematerialization_test.cc | 148 [](const BufferValue& buffer) { return ByteSizeOf(buffer.shape()); }, in RunHloRematerialization()
|
| D | buffer_assignment_test.cc | 1901 static int64 ByteSizeOf(const BufferValue& buffer) { in ByteSizeOf()
2201 ScheduleModule(module.get(), [](const BufferValue& buffer) { in TEST_F()
|
| D | buffer_assignment.cc | 718 BufferValue::ToLocationProto(*alias.instruction(), alias.index()); in ToProto()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_schedule.cc | 205 entry_computation, [pointer_size](const BufferValue& buffer) { in Build()
|