HomeSort by relevance Sort by last modified time
    Searched refs:defining_instruction (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_ordering.cc 103 const HloModule* module = b.defining_instruction()->parent()->parent();
104 if (b.defining_instruction()->parent() == module->entry_computation() &&
105 b.defining_instruction()->opcode() == HloOpcode::kParameter) {
109 if (a.defining_instruction()->parent() == module->entry_computation() &&
110 a.defining_instruction()->opcode() == HloOpcode::kParameter) {
131 v.defining_instruction()->opcode() == HloOpcode::kParameter;
134 call_graph_->InstructionIsNestedIn(b.defining_instruction(),
135 a.defining_instruction()->parent())) {
139 call_graph_->InstructionIsNestedIn(a.defining_instruction(),
140 b.defining_instruction()->parent()))
    [all...]
hlo_value.cc 78 bool equal = defining_instruction() == other.defining_instruction() &&
90 string index_str = defining_instruction()->shape().IsTuple()
94 defining_instruction()->name(), index_str);
hlo_value.h 120 HloInstruction* defining_instruction() const { function in class:xla::HloValue
125 return defining_instruction();
copy_insertion.cc 45 const HloComputation* computation = value.defining_instruction()->parent();
46 return value.defining_instruction()->opcode() == HloOpcode::kParameter &&
49 value.defining_instruction()->parameter_number(), value.index());
53 return value.defining_instruction()->opcode() == HloOpcode::kConstant;
603 const HloInstruction* def = p->value->defining_instruction();
791 CHECK_EQ(copy_value_node->value->defining_instruction()->opcode(),
806 return use->instruction == copy_value_node->value->defining_instruction();
    [all...]
hlo_rematerialization.cc 384 Item* defining_instruction; member in struct:xla::__anon44509::MemoryUsageTracker::Buffer
393 // user of defining_instruction uses this buffer. This can occur due to
406 defining_instruction->instruction->name(), ", size ",
430 CHECK(original_buffer.defining_instruction->placed)
431 << original_buffer.defining_instruction->instruction->name();
448 HloOpcode def_opcode = buffer.defining_instruction->instruction->opcode();
476 return (buffer.defining_instruction->placed &&
481 Buffer& NewBuffer(Item* defining_instruction, int64 size, ItemList&& users,
484 buffers_.push_back(Buffer{buffer_id, defining_instruction, size, live_out,
794 if (buffer.defining_instruction->instruction == instruction)
    [all...]
hlo_alias_analysis.cc 226 if (value.defining_instruction()->opcode() == HloOpcode::kParameter) {
228 value.defining_instruction()->parent();
270 if (value.defining_instruction()->opcode() == HloOpcode::kWhile) {
304 if (value.defining_instruction()->opcode() == HloOpcode::kConditional) {
hlo_dataflow_analysis.cc 84 return value_set.GetUniqueValue().defining_instruction() == instruction;
212 if (value->defining_instruction() == instruction &&
229 current_value->defining_instruction() == instruction &&
    [all...]
bfloat16_propagation.cc     [all...]

Completed in 537 milliseconds