| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_ordering.cc | 145 const HloModule* module = b.defining_instruction()->parent()->parent(); in IsDefinedBefore()
146 if (b.defining_instruction()->parent() == module->entry_computation() && in IsDefinedBefore()
147 b.defining_instruction()->opcode() == HloOpcode::kParameter) { in IsDefinedBefore()
151 if (a.defining_instruction()->parent() == module->entry_computation() && in IsDefinedBefore()
152 a.defining_instruction()->opcode() == HloOpcode::kParameter) { in IsDefinedBefore()
173 v.defining_instruction()->opcode() == HloOpcode::kParameter; in IsDefinedBefore()
176 call_graph_->InstructionIsNestedIn(b.defining_instruction(), in IsDefinedBefore()
177 a.defining_instruction()->parent())) { in IsDefinedBefore()
181 call_graph_->InstructionIsNestedIn(a.defining_instruction(), in IsDefinedBefore()
182 b.defining_instruction()->parent())) { in IsDefinedBefore()
[all …]
|
| D | copy_insertion.cc | 45 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()
212 value->is_phi() && value->defining_instruction() == xla_conditional; in IndicesToCopyForConditional()
498 if (buffer.values().at(0)->defining_instruction()->IsFused()) { in CopyRemover()
621 const HloInstruction* def = p->value->defining_instruction(); in Verify()
817 CHECK_EQ(copy_value_node->value->defining_instruction()->opcode(), in RemoveCopyValue()
832 return use->instruction == copy_value_node->value->defining_instruction(); in RemoveCopyValue()
849 copy_map_.erase(copy_value_node->value->defining_instruction()); in RemoveCopyValue()
[all …]
|
| D | hlo_value.h | 132 HloInstruction* defining_instruction() const { in defining_instruction() function
137 return defining_instruction(); in instruction()
|
| D | hlo_rematerialization.cc | 614 Item* defining_instruction; member
642 defining_instruction->instruction->name(), ", size ", in ToString()
668 CHECK(original_buffer.defining_instruction->placed) in RematerializeBuffer()
669 << original_buffer.defining_instruction->instruction->name(); in RematerializeBuffer()
686 HloInstruction* inst = buffer.defining_instruction->instruction; in AllocatedSize()
713 return (buffer.defining_instruction->placed && in IsCurrentlyLive()
735 Buffer& NewBuffer(Item* defining_instruction, const Shape& shape, in NewBuffer() argument
747 buffer_id, defining_instruction, size_function_(shape), shape, live_out, in NewBuffer()
998 Item* defining_instruction = in MemoryReducedIfRematerialized() local
999 buffers_.at(buffer_id).defining_instruction; in MemoryReducedIfRematerialized()
[all …]
|
| D | hlo_value.cc | 79 bool equal = defining_instruction() == other.defining_instruction() && in operator ==()
|
| D | hlo_alias_analysis.cc | 226 if (value.defining_instruction()->opcode() == HloOpcode::kParameter) { in ComputeWhileAliasedBuffers()
228 value.defining_instruction()->parent(); in ComputeWhileAliasedBuffers()
271 if (value.defining_instruction()->opcode() == HloOpcode::kWhile) { in ComputeWhileAliasedBuffers()
305 if (value.defining_instruction()->opcode() == HloOpcode::kConditional) { in ComputeConditionalAliasedBuffers()
|
| D | loop_schedule_linearizer.cc | 108 HloInstruction* write = value_at_root.defining_instruction(); in AddControlEdgesForLoopWrites()
|
| D | hlo_live_range.cc | 133 if (instruction_schedule_.count(value->defining_instruction()) == 0) { in CalculateBufferStartEndMap()
|
| D | memory_space_assignment.cc | 188 const HloInstruction& defining_instruction = in GetMemoryBoundedness() local
189 *interval.buffer->defining_instruction(); in GetMemoryBoundedness()
191 defining_instruction, in GetMemoryBoundedness()
192 GetInstructionElapsedDueToMemory(defining_instruction, in GetMemoryBoundedness()
824 CHECK_EQ(values_by_defining_inst.count(value.defining_instruction()), 0); in FindAliases()
825 values_by_defining_inst[value.defining_instruction()] = &value; in FindAliases()
1300 instruction_schedule.at(allocation_value.defining_instruction()); in AllocateAllocationValues()
1791 if (value->defining_instruction()->opcode() == HloOpcode::kParameter && in AreIntervalsReservedInAlternateMemory()
1792 value->defining_instruction()->parent() == entry_computation && in AreIntervalsReservedInAlternateMemory()
|
| D | hlo_dataflow_analysis.cc | 129 return value_set.GetUniqueValue().defining_instruction() == instruction; in ValueIsDefinedAt()
278 current_value->defining_instruction() == instruction && in Phi()
1135 if (value->defining_instruction() != instruction) { in Run()
|
| D | memory_space_assignment.h | 755 const HloInstruction* defining_instruction() const { in defining_instruction() function
763 return defining_instruction()->parent(); in computation()
|
| D | memory_space_assignment_test.cc | 1413 a.buffer->defining_instruction()->opcode() == HloOpcode::kMultiply; in TEST_P()
1415 b.buffer->defining_instruction()->opcode() == HloOpcode::kMultiply; in TEST_P()
1423 a.buffer->defining_instruction()->opcode() == HloOpcode::kTanh; in TEST_P()
1425 b.buffer->defining_instruction()->opcode() == HloOpcode::kTanh; in TEST_P()
4027 return get_opcode_priority(a.buffer->defining_instruction()->opcode()) < in TEST_P()
4028 get_opcode_priority(b.buffer->defining_instruction()->opcode()); in TEST_P()
4243 return get_opcode_priority(a.buffer->defining_instruction()->opcode()) < in TEST_P()
4244 get_opcode_priority(b.buffer->defining_instruction()->opcode()); in TEST_P()
4521 return get_opcode_priority(a.buffer->defining_instruction()->opcode()) < in TEST_P()
4522 get_opcode_priority(b.buffer->defining_instruction()->opcode()); in TEST_P()
[all …]
|
| D | bfloat16_propagation.cc | 998 auto hlo = value->defining_instruction(); in ValueTypeAfterChange()
|
| D | heap_simulator.cc | 190 value->defining_instruction()) == 0) { in RunComputation()
|
| /external/deqp-deps/SPIRV-Tools/source/fuzz/ |
| D | fuzzer_util.cpp | 588 auto defining_instruction = context->get_def_use_mgr()->GetDef(id); in IdIsAvailableAtUse() local
593 if (defining_instruction->opcode() == SpvOpFunctionParameter) { in IdIsAvailableAtUse()
594 return InstructionIsFunctionParameter(defining_instruction, in IdIsAvailableAtUse()
601 if (defining_instruction == use_instruction) { in IdIsAvailableAtUse()
619 context->get_instr_block(defining_instruction)->id(), parent_block); in IdIsAvailableAtUse()
621 return dominator_analysis->Dominates(defining_instruction, use_instruction); in IdIsAvailableAtUse()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/fuzz/ |
| D | fuzzer_util.cpp | 588 auto defining_instruction = context->get_def_use_mgr()->GetDef(id); in IdIsAvailableAtUse() local
593 if (defining_instruction->opcode() == SpvOpFunctionParameter) { in IdIsAvailableAtUse()
594 return InstructionIsFunctionParameter(defining_instruction, in IdIsAvailableAtUse()
601 if (defining_instruction == use_instruction) { in IdIsAvailableAtUse()
619 context->get_instr_block(defining_instruction)->id(), parent_block); in IdIsAvailableAtUse()
621 return dominator_analysis->Dominates(defining_instruction, use_instruction); in IdIsAvailableAtUse()
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/source/fuzz/ |
| D | fuzzer_util.cpp | 639 auto defining_instruction = context->get_def_use_mgr()->GetDef(id); in IdIsAvailableAtUse() local
644 if (defining_instruction->opcode() == SpvOpFunctionParameter) { in IdIsAvailableAtUse()
645 return InstructionIsFunctionParameter(defining_instruction, in IdIsAvailableAtUse()
652 if (defining_instruction == use_instruction) { in IdIsAvailableAtUse()
669 context->get_instr_block(defining_instruction)->id(), parent_block); in IdIsAvailableAtUse()
671 return dominator_analysis->Dominates(defining_instruction, use_instruction); in IdIsAvailableAtUse()
|