| /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_canonicalizer.cc | 25 TF_RET_CHECK(conditional->opcode() == HloOpcode::kConditional); in CanonicalizeNonTupleConditional()
52 if (inst->opcode() == HloOpcode::kConditional && in Run()
|
| D | bfloat16_propagation.cc | 210 CHECK_EQ(cond->opcode(), HloOpcode::kConditional); in DetermineConditionalComputationsPrecision()
297 } else if (use.instruction->opcode() == HloOpcode::kConditional) { in AllUsersConsumeBF16()
383 } else if (hlo->opcode() == HloOpcode::kConditional) { in DetermineInstructionPrecision()
397 if (hlo->opcode() == HloOpcode::kConditional && in DetermineInstructionPrecision()
509 case HloOpcode::kConditional: in AdjustCalledComputationParameters()
565 case HloOpcode::kConditional: in AdjustCalledComputationRoot()
686 } else if (hlo->opcode() == HloOpcode::kConditional) { in ResolveInconsistencyOfAliasingBuffersHelper()
|
| D | hlo_opcode_test.cc | 57 case HloOpcode::kConditional: in TEST()
|
| D | conditional_simplifier_test.cc | 137 ASSERT_EQ(conditional->opcode(), HloOpcode::kConditional); in TEST_F()
153 ASSERT_EQ(conditional->opcode(), HloOpcode::kConditional); in TEST_F()
167 ASSERT_EQ(conditional->opcode(), HloOpcode::kConditional); in TEST_F()
|
| D | hlo_ordering.cc | 94 (a_ancestor->opcode() == HloOpcode::kConditional)) { in GetExecutionConstraint()
199 b.defining_instruction()->opcode() == HloOpcode::kConditional) { in IsDefinedBefore()
318 if (use.instruction->opcode() == HloOpcode::kConditional) { in UsesBeforeValueDefinition()
|
| D | conditional_to_select.cc | 76 if (callsite.instruction()->opcode() == HloOpcode::kConditional) { in Run()
|
| D | hlo_dataflow_analysis.cc | 56 instruction->opcode() == HloOpcode::kConditional) { in CalculatePostOrderScheduleHelper()
515 CHECK_EQ(conditional->opcode(), HloOpcode::kConditional); in UpdateConditionalValueSet()
645 } else if (callsite.instruction()->opcode() == HloOpcode::kConditional) { in UpdateParameterValueSet()
805 case HloOpcode::kConditional: in UpdateInstructionValueSet()
873 if (user->opcode() == HloOpcode::kConditional) { in Propagate()
908 callsite.instruction()->opcode() == HloOpcode::kConditional) { in Propagate()
970 case HloOpcode::kConditional: in InitializeInstructionValueSets()
|
| D | flatten_call_graph.cc | 51 case HloOpcode::kConditional: { in ReplaceCalledComputation()
|
| D | hlo_opcode.h | 72 V(kConditional, "conditional", kHloOpcodeIsVariadic) \
|
| D | hlo_instruction.cc | 236 case HloOpcode::kConditional: { in CreateFromProto()
1203 absl::WrapUnique(new HloInstruction(HloOpcode::kConditional, shape)); in CreateConditional()
1220 absl::WrapUnique(new HloInstruction(HloOpcode::kConditional, shape)); in CreateConditional()
1797 case HloOpcode::kConditional: in CloneWithNewOperands()
2185 case HloOpcode::kConditional: in IdenticalSlowPath()
2500 CHECK_EQ(HloOpcode::kConditional, opcode_); in true_computation()
2506 CHECK_EQ(HloOpcode::kConditional, opcode_); in false_computation()
2513 CHECK(HloOpcode::kConditional == opcode_); in branch_computations()
2518 CHECK(HloOpcode::kConditional == opcode_); in branch_count()
2523 CHECK(HloOpcode::kConditional == opcode_); in branch_computation()
[all …]
|
| D | conditional_simplifier.cc | 71 CHECK_EQ(conditional->opcode(), HloOpcode::kConditional); in TryRemoveConditional()
583 if (instr->opcode() == HloOpcode::kConditional) { in Run()
|
| D | sharding_propagation.cc | 241 case HloOpcode::kConditional: in PickRepresentativeOperand()
301 case HloOpcode::kConditional: in SupportSpatialPartitioning()
869 case HloOpcode::kConditional: { in InferShardingFromOperands()
1677 } else if (inst->opcode() == HloOpcode::kConditional) { in Run()
1710 if (instruction->opcode() == HloOpcode::kConditional || in Run()
1738 instruction->opcode() == HloOpcode::kConditional) { in Run()
|
| D | hlo_element_type_converter.cc | 149 opcode == HloOpcode::kSort || opcode == HloOpcode::kConditional) { in Run()
|
| D | bfloat16_conversion_folding.cc | 175 hlo->opcode() == HloOpcode::kConditional || // in DefaultAction()
|
| D | hlo_live_range.cc | 105 instruction->opcode() == HloOpcode::kConditional) { in FlattenSchedule()
|
| D | hlo_module_group_metadata.cc | 323 } else if (hlo->opcode() == HloOpcode::kConditional) { in RecordInstructions()
403 instruction1->opcode() == HloOpcode::kConditional || in AddCompanion()
|
| D | ar_crs_combiner.cc | 213 if (caller_instruction->opcode() == HloOpcode::kConditional) { in ConditionalFromBodyParameter()
289 case HloOpcode::kConditional: { in GetAllTuples()
|
| D | hlo_alias_analysis.cc | 288 if (callsite.instruction()->opcode() == HloOpcode::kConditional) { in ComputeConditionalAliasedBuffers()
305 if (value.defining_instruction()->opcode() == HloOpcode::kConditional) { in ComputeConditionalAliasedBuffers()
|
| D | layout_assignment.cc | 607 } else if (instruction->opcode() == HloOpcode::kConditional) { in AddMandatoryConstraints()
878 if (instruction->opcode() == HloOpcode::kConditional && operand_no > 0 && in CopyOperandIfLayoutsDiffer()
996 case HloOpcode::kConditional: { in CheckLayouts()
2124 return caller.instruction()->opcode() == HloOpcode::kConditional; in Run()
2130 if (caller->opcode() == HloOpcode::kConditional) { in Run()
2219 case HloOpcode::kConditional: in InstructionCanChangeLayout()
|
| D | hlo_liveness_analysis.cc | 240 } else if (caller->opcode() == HloOpcode::kConditional) { in PropagateLivenessThroughControlFlow()
|
| D | memory_space_assignment.cc | 204 use.instruction->opcode() == HloOpcode::kConditional) { in GetMemoryBoundedness()
412 instruction->opcode() == HloOpcode::kConditional) { in CostAnalysisPrefetchIntervalPicker()
944 } else if (use.instruction->opcode() == HloOpcode::kConditional) { in IsUseAllowedInAlternateMemory()
1260 if (position.instruction->opcode() == HloOpcode::kConditional) { in AddRequiredAssignmentsForColocatedIntervals()
1362 } else if (hlo_use.instruction->opcode() == HloOpcode::kConditional) { in AllocateAllocationValues()
1389 HloOpcode::kConditional && in AllocateAllocationValues()
3392 if (last_use_instruction->opcode() == HloOpcode::kConditional) { in VerifyAndExportHeapSimulatorTrace()
3414 last_use_instruction->opcode() == HloOpcode::kConditional) { in VerifyAndExportHeapSimulatorTrace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk.cc | 39 case Thunk::kConditional: in ThunkKindToString()
|
| D | conditional_thunk.cc | 51 : Thunk(Kind::kConditional, thunk_info), in ConditionalThunk()
|
| D | thunk.h | 49 kConditional, enumerator
|