Home
last modified time | relevance | path

Searched refs:defining_instruction (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering.cc103 const HloModule* module = b.defining_instruction()->parent()->parent(); in IsDefinedBefore()
104 if (b.defining_instruction()->parent() == module->entry_computation() && in IsDefinedBefore()
105 b.defining_instruction()->opcode() == HloOpcode::kParameter) { in IsDefinedBefore()
109 if (a.defining_instruction()->parent() == module->entry_computation() && in IsDefinedBefore()
110 a.defining_instruction()->opcode() == HloOpcode::kParameter) { in IsDefinedBefore()
131 v.defining_instruction()->opcode() == HloOpcode::kParameter; in IsDefinedBefore()
134 call_graph_->InstructionIsNestedIn(b.defining_instruction(), in IsDefinedBefore()
135 a.defining_instruction()->parent())) { in IsDefinedBefore()
139 call_graph_->InstructionIsNestedIn(a.defining_instruction(), in IsDefinedBefore()
140 b.defining_instruction()->parent())) { in IsDefinedBefore()
[all …]
Dhlo_value.cc78 bool equal = defining_instruction() == other.defining_instruction() && in operator ==()
90 string index_str = defining_instruction()->shape().IsTuple() in ToShortString()
94 defining_instruction()->name(), index_str); in ToShortString()
Dcopy_insertion.cc45 const HloComputation* computation = value.defining_instruction()->parent(); in IsReadonlyEntryParameterValue()
46 return value.defining_instruction()->opcode() == HloOpcode::kParameter && in IsReadonlyEntryParameterValue()
49 value.defining_instruction()->parameter_number(), value.index()); in IsReadonlyEntryParameterValue()
53 return value.defining_instruction()->opcode() == HloOpcode::kConstant; in IsConstantValue()
603 const HloInstruction* def = p->value->defining_instruction(); in Verify()
791 CHECK_EQ(copy_value_node->value->defining_instruction()->opcode(), in RemoveCopyValue()
806 return use->instruction == copy_value_node->value->defining_instruction(); in RemoveCopyValue()
823 copy_map_.erase(copy_value_node->value->defining_instruction()); in RemoveCopyValue()
1020 add_index_to_copy(value->defining_instruction(), value->defining_index()); in AddSpecialCaseCopies()
1263 HloInstruction* def = value->defining_instruction(); in AddCopiesForBufferAssignment()
Dhlo_rematerialization.cc384 Item* defining_instruction; member
406 defining_instruction->instruction->name(), ", size ", in ToString()
430 CHECK(original_buffer.defining_instruction->placed) in RematerializeBuffer()
431 << original_buffer.defining_instruction->instruction->name(); in RematerializeBuffer()
448 HloOpcode def_opcode = buffer.defining_instruction->instruction->opcode(); in AllocatedSize()
476 return (buffer.defining_instruction->placed && in IsCurrentlyLive()
481 Buffer& NewBuffer(Item* defining_instruction, int64 size, ItemList&& users, in NewBuffer() argument
484 buffers_.push_back(Buffer{buffer_id, defining_instruction, size, live_out, in NewBuffer()
794 if (buffer.defining_instruction->instruction == instruction) { in Check()
Dhlo_value.h120 HloInstruction* defining_instruction() const { in defining_instruction() function
125 return defining_instruction(); in instruction()
Dhlo_alias_analysis.cc226 if (value.defining_instruction()->opcode() == HloOpcode::kParameter) { in ComputeWhileAliasedBuffers()
228 value.defining_instruction()->parent(); in ComputeWhileAliasedBuffers()
270 if (value.defining_instruction()->opcode() == HloOpcode::kWhile) { in ComputeWhileAliasedBuffers()
304 if (value.defining_instruction()->opcode() == HloOpcode::kConditional) { in ComputeConditionalAliasedBuffers()
Dhlo_dataflow_analysis.cc84 return value_set.GetUniqueValue().defining_instruction() == instruction; in ValueIsDefinedAt()
212 if (value->defining_instruction() == instruction && in Phi()
229 current_value->defining_instruction() == instruction && in Phi()
865 if (value->defining_instruction() != instruction) { in Run()
Dbfloat16_propagation.cc844 auto hlo = value->defining_instruction(); in ValueTypeAfterChange()