Searched refs:defining_position (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment.cc | 197 interval.buffer->defining_position().instruction, in GetMemoryBoundedness()
198 interval.buffer->defining_position().index)) { in GetMemoryBoundedness()
708 return defining_position() == other.defining_position() && in operator ==()
800 allocation_value->defining_position().instruction) < use_time) { in CreateAllocationValues()
833 use->aliases.push_back(aliased_value_it->second->defining_position()); in FindAliases()
1021 instruction_schedule.at(interval.buffer->defining_position().instruction); in AppendBufferInfoDebugString()
1547 buffer->defining_position(), MemorySpace::kDefault, kDummyChunk, in AllocateCrossProgramPrefetchBuffer()
1574 buffer->defining_position().shape(), last_use_time, in AllocateCrossProgramPrefetchBuffer()
1793 is_position_in_alternate_memory(value->defining_position())) { in AreIntervalsReservedInAlternateMemory()
1953 const HloPosition& defining_position = in AllocateSegment() local
[all …]
|
| D | hlo_value.h | 129 const HloPosition& defining_position() const { return positions_[0]; } in defining_position() function
133 return defining_position().instruction; in defining_instruction()
142 const ShapeIndex& defining_index() const { return defining_position().index; } in defining_index()
147 const Shape& shape() const override { return defining_position().shape(); } in shape()
|
| D | buffer_assignment.h | 592 const HloPosition& defining_position = value->defining_position(); in DefaultColorer()
593 if (defining_position.shape().has_layout()) { in DefaultColorer()
595 defining_position.shape().layout().memory_space())); in DefaultColorer()
|
| D | memory_space_assignment.h | 524 Allocation(HloPosition defining_position, MemorySpace memory_space, in Allocation() argument
526 : defining_position_(defining_position), in Allocation()
547 virtual HloPosition defining_position() const { return defining_position_; } in defining_position() function
604 HloPosition defining_position() const override { in defining_position() function
611 return prev_allocation_.defining_position(); in defining_position()
754 const HloPosition& defining_position() const { return defining_position_; } in defining_position() function
756 return defining_position().instruction; in defining_instruction()
|
| D | hlo_value.cc | 159 DCHECK_NE(position_a, defining_position()); in SetPositionsAndComputeUses()
|
| D | buffer_assignment.cc | 854 instruction_schedule.at(buffer.defining_position().instruction); in BufferInfoString()
1383 const HloPosition& defining_position = position_and_chunk.first; in AssignPresetBuffers() local
1385 defining_position.instruction, defining_position.index); in AssignPresetBuffers()
|
| D | bfloat16_propagation.cc | 999 const auto& position = value->defining_position(); in ValueTypeAfterChange()
|
| D | layout_assignment.cc | 2017 const Shape& defining_shape = value->defining_position().shape(); in PropagateMemorySpace()
|
| D | memory_space_assignment_test.cc | 245 if (pos_and_chunk.first == value->defining_position()) {
|