| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_support.cc | 27 case HloOpcode::kConditional: in SupportsBF16Operand()
46 case HloOpcode::kConditional: in SupportsBF16Output()
64 case HloOpcode::kConditional: in SupportsMixedPrecisions()
|
| D | conditional_simplifier_test.cc | 122 ASSERT_EQ(conditional->opcode(), HloOpcode::kConditional); in TEST_F()
138 ASSERT_EQ(conditional->opcode(), HloOpcode::kConditional); in TEST_F()
152 ASSERT_EQ(conditional->opcode(), HloOpcode::kConditional); in TEST_F()
|
| D | conditional_simplifier.cc | 42 CHECK_EQ(conditional->opcode(), HloOpcode::kConditional); in TryRemoveConditional()
95 if (instr->opcode() == HloOpcode::kConditional) { in Run()
|
| D | hlo_ordering.cc | 75 (a_ancestor->opcode() == HloOpcode::kConditional)) { in ExecutesBefore()
157 b.defining_instruction()->opcode() == HloOpcode::kConditional) { in IsDefinedBefore()
233 if (use.instruction->opcode() == HloOpcode::kConditional) { in UseIsBeforeValueDefinition()
|
| D | hlo_dataflow_analysis.cc | 376 CHECK_EQ(conditional->opcode(), HloOpcode::kConditional); in UpdateConditionalValueSet()
506 } else if (callsite.instruction()->opcode() == HloOpcode::kConditional) { in UpdateParameterValueSet()
626 case HloOpcode::kConditional: in UpdateInstructionValueSet()
674 if (user->opcode() == HloOpcode::kConditional) { in Propagate()
709 callsite.instruction()->opcode() == HloOpcode::kConditional) { in Propagate()
778 case HloOpcode::kConditional: in InitializeInstructionValueSets()
|
| D | hlo_opcode_test.cc | 57 case HloOpcode::kConditional: in TEST()
|
| D | flatten_call_graph.cc | 51 case HloOpcode::kConditional: { in ReplaceCalledComputation()
|
| D | hlo_instruction.cc | 214 case HloOpcode::kConditional: { in CreateFromProto()
957 absl::WrapUnique(new HloInstruction(HloOpcode::kConditional, shape)); in CreateConditional()
974 absl::WrapUnique(new HloInstruction(HloOpcode::kConditional, shape)); in CreateConditional()
1477 case HloOpcode::kConditional: in CloneWithNewOperands()
1785 case HloOpcode::kConditional: in IdenticalSlowPath()
2058 CHECK_EQ(HloOpcode::kConditional, opcode_); in true_computation()
2064 CHECK_EQ(HloOpcode::kConditional, opcode_); in false_computation()
2071 CHECK(HloOpcode::kConditional == opcode_); in branch_computations()
2076 CHECK(HloOpcode::kConditional == opcode_); in branch_count()
2081 CHECK(HloOpcode::kConditional == opcode_); in branch_computation()
[all …]
|
| D | hlo_opcode.h | 73 V(kConditional, "conditional", kHloOpcodeIsVariadic) \
|
| D | bfloat16_conversion_folding.cc | 165 hlo->opcode() == HloOpcode::kConditional || // in DefaultAction()
|
| D | hlo_module_group_metadata.cc | 306 } else if (hlo->opcode() == HloOpcode::kConditional) { in RecordInstructions()
385 instruction1->opcode() == HloOpcode::kConditional || in AddCompanion()
|
| D | hlo_alias_analysis.cc | 287 if (callsite.instruction()->opcode() == HloOpcode::kConditional) { in ComputeConditionalAliasedBuffers()
304 if (value.defining_instruction()->opcode() == HloOpcode::kConditional) { in ComputeConditionalAliasedBuffers()
|
| D | hlo_element_type_converter.cc | 149 opcode == HloOpcode::kSort || opcode == HloOpcode::kConditional) { in Run()
|
| D | hlo_liveness_analysis.cc | 240 } else if (caller->opcode() == HloOpcode::kConditional) { in PropagateLivenessThroughControlFlow()
|
| D | call_graph.cc | 58 case HloOpcode::kConditional: in GetInstructionCallContext()
|
| D | hlo_verifier.cc | 45 case HloOpcode::kConditional: in IsCallerInstruction()
763 case HloOpcode::kConditional: in CheckMixedPrecisionOperands()
867 case HloOpcode::kConditional: in CheckShape()
|
| D | heap_simulator.cc | 297 instruction->opcode() == HloOpcode::kConditional || in RunComputation()
547 instruction->opcode() == HloOpcode::kConditional)) { in AccountForSubcomputationMemory()
|
| D | bfloat16_normalization.cc | 386 hlo->opcode() == HloOpcode::kConditional || // in DefaultAction()
|
| D | copy_insertion.cc | 321 TF_RET_CHECK(conditional->opcode() == HloOpcode::kConditional); in AddCopiesForConditional()
973 } else if (instruction->opcode() == HloOpcode::kConditional) { in AddCopiesToResolveInterference()
|
| D | instruction_fusion.cc | 125 case HloOpcode::kConditional: in IsExpensive()
|
| D | hlo_memory_scheduler.cc | 245 opcode == HloOpcode::kConditional)) { in BytesFreedIfScheduled()
|
| D | layout_assignment.cc | 589 } else if (instruction->opcode() == HloOpcode::kConditional) { in AddMandatoryConstraints()
940 case HloOpcode::kConditional: { in CheckLayouts()
1993 case HloOpcode::kConditional: in InstructionCanChangeLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk.cc | 25 case Thunk::kConditional: in ThunkKindToString()
|
| D | thunk.h | 46 kConditional, enumerator
|
| D | conditional_thunk.cc | 31 : Thunk(Kind::kConditional, hlo), in ConditionalThunk()
|