| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_ordering.cc | 146 const HloModule* module = b.defining_instruction()->parent()->parent(); in IsDefinedBefore()
147 if (b.defining_instruction()->parent() == module->entry_computation() && in IsDefinedBefore()
148 b.defining_instruction()->opcode() == HloOpcode::kParameter) { in IsDefinedBefore()
152 if (a.defining_instruction()->parent() == module->entry_computation() && in IsDefinedBefore()
153 a.defining_instruction()->opcode() == HloOpcode::kParameter) { in IsDefinedBefore()
174 v.defining_instruction()->opcode() == HloOpcode::kParameter; in IsDefinedBefore()
177 call_graph_->InstructionIsNestedIn(b.defining_instruction(), in IsDefinedBefore()
178 a.defining_instruction()->parent())) { in IsDefinedBefore()
182 call_graph_->InstructionIsNestedIn(a.defining_instruction(), in IsDefinedBefore()
183 b.defining_instruction()->parent())) { in IsDefinedBefore()
[all …]
|
| D | hlo_rematerialization.cc | 619 Item* defining_instruction; member
647 defining_instruction->instruction->name(), ", size ", in ToString()
673 CHECK(original_buffer.defining_instruction->placed) in RematerializeBuffer()
674 << original_buffer.defining_instruction->instruction->name(); in RematerializeBuffer()
691 HloInstruction* inst = buffer.defining_instruction->instruction; in AllocatedSize()
718 return (buffer.defining_instruction->placed && in IsCurrentlyLive()
740 Buffer& NewBuffer(Item* defining_instruction, const Shape& shape, in NewBuffer() argument
752 buffer_id, defining_instruction, size_function_(shape), shape, live_out, in NewBuffer()
1003 Item* defining_instruction = in MemoryReducedIfRematerialized() local
1004 buffers_.at(buffer_id).defining_instruction; in MemoryReducedIfRematerialized()
[all …]
|
| D | hlo_value.h | 132 HloInstruction* defining_instruction() const { in defining_instruction() function
137 return defining_instruction(); in instruction()
|
| D | copy_insertion.cc | 50 const HloComputation* computation = value.defining_instruction()->parent(); in IsReadonlyEntryParameterValue()
51 return value.defining_instruction()->opcode() == HloOpcode::kParameter && in IsReadonlyEntryParameterValue()
54 value.defining_instruction()->parameter_number(), value.index()); in IsReadonlyEntryParameterValue()
58 return value.defining_instruction()->opcode() == HloOpcode::kConstant; in IsConstantValue()
217 value->is_phi() && value->defining_instruction() == xla_conditional; in IndicesToCopyForConditional()
1148 if (buffer.values().at(0)->defining_instruction()->IsFused()) { in CopyRemover()
1275 const HloInstruction* def = p->value->defining_instruction(); in Verify()
1499 CHECK_EQ(copy_value_node->value->defining_instruction()->opcode(), in RemoveCopyValue()
1514 return use->instruction == copy_value_node->value->defining_instruction(); in RemoveCopyValue()
1531 copy_map_.erase(copy_value_node->value->defining_instruction()); in RemoveCopyValue()
[all …]
|
| 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 | hlo_value.cc | 79 bool equal = defining_instruction() == other.defining_instruction() && in operator ==()
|
| D | memory_space_assignment.cc | 200 const HloInstruction& defining_instruction = in GetMemoryBoundedness() local
201 *interval.buffer->defining_instruction(); in GetMemoryBoundedness()
203 defining_instruction, in GetMemoryBoundedness()
205 defining_instruction, in GetMemoryBoundedness()
825 if (allocation_value->defining_instruction() == in CreateAllocationValues()
830 allocation_value->defining_instruction()->opcode()) && in CreateAllocationValues()
845 allocation_value.defining_instruction()->opcode())) { in CreateAllocationValues()
864 values_by_defining_inst[value.defining_instruction()].push_back(&value); in FindAliases()
1335 instruction_vector.push_back(allocation_value.defining_instruction()); in CreateAllocationValuesFromColocatedIntervals()
1384 instruction_schedule.at(allocation_value.defining_instruction()); in AllocateAllocationValues()
[all …]
|
| D | loop_schedule_linearizer.cc | 108 HloInstruction* write = value_at_root.defining_instruction(); in AddControlEdgesForLoopWrites()
|
| D | hlo_live_range.cc | 135 if (instruction_schedule_.count(value->defining_instruction()) == 0) { in CalculateBufferStartEndMap()
|
| D | memory_space_assignment_test.cc | 1432 a.buffer->defining_instruction()->opcode() == HloOpcode::kMultiply; in TEST_P()
1434 b.buffer->defining_instruction()->opcode() == HloOpcode::kMultiply; in TEST_P()
1442 a.buffer->defining_instruction()->opcode() == HloOpcode::kTanh; in TEST_P()
1444 b.buffer->defining_instruction()->opcode() == HloOpcode::kTanh; in TEST_P()
4046 return get_opcode_priority(a.buffer->defining_instruction()->opcode()) < in TEST_P()
4047 get_opcode_priority(b.buffer->defining_instruction()->opcode()); in TEST_P()
4264 return get_opcode_priority(a.buffer->defining_instruction()->opcode()) < in TEST_P()
4265 get_opcode_priority(b.buffer->defining_instruction()->opcode()); in TEST_P()
5215 return get_opcode_priority(a.buffer->defining_instruction()->opcode()) < in TEST_P()
5216 get_opcode_priority(b.buffer->defining_instruction()->opcode()); in TEST_P()
[all …]
|
| D | memory_space_assignment.h | 766 const HloInstruction* defining_instruction() const { in defining_instruction() function
774 return defining_instruction()->parent(); in computation()
|
| D | hlo_dataflow_analysis.cc | 298 return value_set.GetUniqueValue().defining_instruction() == instruction; in ValueIsDefinedAt()
447 current_value->defining_instruction() == instruction && in Phi()
1474 if (value->defining_instruction() != instruction) { in Run()
|
| D | bfloat16_propagation.cc | 993 auto hlo = value->defining_instruction(); in ValueTypeAfterChange()
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/source/fuzz/ |
| D | fuzzer_util.cpp | 667 auto defining_instruction = context->get_def_use_mgr()->GetDef(id); in IdIsAvailableAtUse() local
672 if (defining_instruction->opcode() == SpvOpFunctionParameter) { in IdIsAvailableAtUse()
673 return InstructionIsFunctionParameter(defining_instruction, in IdIsAvailableAtUse()
680 if (defining_instruction == use_instruction) { in IdIsAvailableAtUse()
697 context->get_instr_block(defining_instruction)->id(), parent_block); in IdIsAvailableAtUse()
699 return dominator_analysis->Dominates(defining_instruction, use_instruction); in IdIsAvailableAtUse()
|
| /external/deqp-deps/SPIRV-Tools/source/fuzz/ |
| D | fuzzer_util.cpp | 667 auto defining_instruction = context->get_def_use_mgr()->GetDef(id); in IdIsAvailableAtUse() local
672 if (defining_instruction->opcode() == SpvOpFunctionParameter) { in IdIsAvailableAtUse()
673 return InstructionIsFunctionParameter(defining_instruction, in IdIsAvailableAtUse()
680 if (defining_instruction == use_instruction) { in IdIsAvailableAtUse()
697 context->get_instr_block(defining_instruction)->id(), parent_block); in IdIsAvailableAtUse()
699 return dominator_analysis->Dominates(defining_instruction, use_instruction); in IdIsAvailableAtUse()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/fuzz/ |
| D | fuzzer_util.cpp | 667 auto defining_instruction = context->get_def_use_mgr()->GetDef(id); in IdIsAvailableAtUse() local
672 if (defining_instruction->opcode() == SpvOpFunctionParameter) { in IdIsAvailableAtUse()
673 return InstructionIsFunctionParameter(defining_instruction, in IdIsAvailableAtUse()
680 if (defining_instruction == use_instruction) { in IdIsAvailableAtUse()
697 context->get_instr_block(defining_instruction)->id(), parent_block); in IdIsAvailableAtUse()
699 return dominator_analysis->Dominates(defining_instruction, use_instruction); in IdIsAvailableAtUse()
|