Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_graph_dumper.cc533 if (from_node->IsFused() && in Header()
538 if (to_node->IsFused() && to_node->opcode() == HloOpcode::kParameter) { in Header()
719 if (instr->opcode() != HloOpcode::kParameter || !instr->IsFused()) { in TryGetFusionParameterConstant()
748 !instr->IsFused() && in ShouldMergeIntoUsers()
985 if (instr->IsFused() && in GetInstructionColor()
1000 if (ShapeUtil::IsEffectiveScalar(instr->shape()) || instr->IsFused()) { in GetInstructionColor()
1207 if (instr->opcode() == HloOpcode::kParameter && instr->IsFused()) { in AddInstructionIncomingEdges()
Dhlo_instruction.cc2008 CHECK(!IsFused()); in set_to_apply()
2039 CHECK(!IsFused()); in set_while_condition()
2047 CHECK(!IsFused()); in set_while_body()
2091 CHECK(!IsFused()); in set_branch_computation()
2450 bool HloInstruction::IsFused() const { return parent_->IsFusionComputation(); } in IsFused() function in xla::HloInstruction
Dhlo_instructions.h1140 CHECK(!IsFused()); in set_select()
1147 CHECK(!IsFused()); in set_scatter()
Dtuple_points_to_analysis.cc576 const string prefix = instruction->IsFused() ? " " : ""; in InstructionToString()
Dhlo_instruction.h1131 bool IsFused() const;
Delemental_ir_emitter.cc1594 CHECK(!hlo->IsFused()); in MakePhiloxRngElementGenerator()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc199 while (c->IsFused()) { in BuffersInvariantWithinConsumer()
Dir_emitter_unnested.cc1178 scatter->IsFused() ? scatter->parent()->FusionInstruction() : scatter; in EmitScatter()