Searched refs:hlo_buffer (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.cc | 409 const HloBuffer& hlo_buffer) const { in GetAssignedAllocation()
410 return GetAssignedAllocation(*hlo_buffer.values()[0]); in GetAssignedAllocation()
985 const HloBuffer& hlo_buffer, in MaybeAssignBuffer() argument
987 CHECK(!assignment->HasAllocation(hlo_buffer)) in MaybeAssignBuffer()
988 << "buffer " << hlo_buffer << " already has an allocation assigned."; in MaybeAssignBuffer()
990 VLOG(4) << "Trying to assign " << hlo_buffer << " size " in MaybeAssignBuffer()
991 << assignment->HloBufferSize(hlo_buffer) in MaybeAssignBuffer()
994 if (hlo_buffer.color() != allocation->color()) { in MaybeAssignBuffer()
995 VLOG(4) << "Can't assign: buffer has color " << hlo_buffer.color() in MaybeAssignBuffer()
1000 if (assignment->HloBufferSize(hlo_buffer) > allocation->size()) { in MaybeAssignBuffer()
[all …]
|
| D | hlo_live_range.cc | 35 for (const HloBuffer& hlo_buffer : alias_analysis_.buffers()) { in NormalizeAliasedBuffers() local
37 for (const HloValue* hlo_value : hlo_buffer.values()) { in NormalizeAliasedBuffers()
|
| D | heap_simulator.cc | 233 const HloBuffer* hlo_buffer = in RunComputation() local
235 if (first_allocated_value.count(hlo_buffer) != 0) { in RunComputation()
240 ShareBuffer(value, first_allocated_value[hlo_buffer], in RunComputation()
243 << first_allocated_value[hlo_buffer]->ToShortString(); in RunComputation()
247 hlo_buffer->values().size() == 1) { in RunComputation()
318 first_allocated_value[hlo_buffer] = value; in RunComputation()
|
| D | buffer_assignment.h | 389 const HloBuffer& hlo_buffer) const;
663 const HloBuffer* hlo_buffer, bool is_thread_local,
|
| D | BUILD | 330 ":hlo_buffer",
1316 ":hlo_buffer",
1421 ":hlo_buffer",
1444 ":hlo_buffer",
3375 name = "hlo_buffer",
3376 srcs = ["hlo_buffer.cc"],
3377 hdrs = ["hlo_buffer.h"],
3398 ":hlo_buffer",
|