| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_support.cc | 32 case HloOpcode::kWhile: in SupportsBF16Operand()
51 case HloOpcode::kWhile: in SupportsBF16Output()
69 case HloOpcode::kWhile: in SupportsMixedPrecisions()
|
| D | while_loop_simplifier_test.cc | 41 return instr->opcode() == HloOpcode::kWhile; in FindFirstWhile()
169 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile); in TEST_F()
186 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile); in TEST_F()
202 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile); in TEST_F()
218 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile); in TEST_F()
233 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile); in TEST_F()
426 return (instr->opcode() == HloOpcode::kWhile && in TEST_F()
481 return (instr->opcode() == HloOpcode::kWhile && in TEST_F()
|
| D | while_util_test.cc | 78 ASSERT_EQ(while_instr->opcode(), HloOpcode::kWhile); in TEST_F()
109 ASSERT_EQ(while_instr->opcode(), HloOpcode::kWhile); in TEST_F()
213 return instr->opcode() == HloOpcode::kWhile; in TEST_F()
|
| D | hlo_liveness_analysis.cc | 173 CHECK_EQ(instruction->opcode(), HloOpcode::kWhile); in PropagateLivenessThroughWhile()
203 if (callsite.instruction()->opcode() == HloOpcode::kWhile) { in PropagateLivenessToParameterCallers()
234 if (caller->opcode() == HloOpcode::kWhile) { in PropagateLivenessThroughControlFlow()
288 } else if (instruction->opcode() == HloOpcode::kWhile) { in RunAnalysis()
|
| D | gather_expander_test.cc | 81 if (instr->opcode() == HloOpcode::kWhile) { in TEST_F()
144 if (instr->opcode() == HloOpcode::kWhile) { in TEST_F()
|
| D | bfloat16_propagation.cc | 167 CHECK_EQ(while_hlo->opcode(), HloOpcode::kWhile); in DetermineWhileComputationsPrecision()
283 } else if (use.instruction->opcode() == HloOpcode::kWhile) { in AllUsersConsumeBF16()
381 } else if (hlo->opcode() == HloOpcode::kWhile) { in DetermineInstructionPrecision()
390 if (hlo->opcode() == HloOpcode::kWhile && in DetermineInstructionPrecision()
505 case HloOpcode::kWhile: in AdjustCalledComputationParameters()
562 case HloOpcode::kWhile: in AdjustCalledComputationRoot()
663 if (hlo->opcode() == HloOpcode::kWhile) { in ResolveInconsistencyOfAliasingBuffersHelper()
854 if (inst->opcode() != HloOpcode::kWhile) { in Run()
|
| D | hlo_ordering.cc | 79 if (a_ancestor == b_ancestor && a_ancestor->opcode() == HloOpcode::kWhile) { in GetExecutionConstraint()
188 if (b.is_phi() && b.defining_instruction()->opcode() == HloOpcode::kWhile && in IsDefinedBefore()
271 if (use.instruction->opcode() == HloOpcode::kWhile) { in UsesBeforeValueDefinition()
295 if (value.defining_instruction()->opcode() == HloOpcode::kWhile) { in UsesBeforeValueDefinition()
|
| D | while_loop_trip_count_annotator.cc | 26 if (instr->opcode() != HloOpcode::kWhile) { in Run()
|
| D | hlo_module_dce_test.cc | 50 if (instruction->opcode() == HloOpcode::kWhile && in WhileBodyHasPassThroughTupleElement()
75 if (instruction->opcode() == HloOpcode::kWhile) { in GetWhileLoops()
|
| D | hlo_live_range.cc | 111 if (instruction->opcode() == HloOpcode::kWhile) { in FlattenSchedule()
145 if (module_scoped_analysis_ && used->opcode() == HloOpcode::kWhile) { in CalculateBufferStartEndMap()
|
| D | hlo_alias_analysis.cc | 215 if (use.instruction->opcode() == HloOpcode::kWhile) { in ComputeWhileAliasedBuffers()
232 if (callsite.instruction()->opcode() == HloOpcode::kWhile) { in ComputeWhileAliasedBuffers()
252 if (callsite.instruction()->opcode() == HloOpcode::kWhile && in ComputeWhileAliasedBuffers()
271 if (value.defining_instruction()->opcode() == HloOpcode::kWhile) { in ComputeWhileAliasedBuffers()
|
| D | hlo_dataflow_analysis.cc | 63 if (instruction->opcode() == HloOpcode::kWhile) { in CalculatePostOrderScheduleHelper()
629 } else if (callsite.instruction()->opcode() == HloOpcode::kWhile) { in UpdateParameterValueSet()
725 CHECK_EQ(xla_while->opcode(), HloOpcode::kWhile); in UpdateWhileValueSet()
795 case HloOpcode::kWhile: in UpdateInstructionValueSet()
910 } else if (callsite.instruction()->opcode() == HloOpcode::kWhile) { in Propagate()
968 case HloOpcode::kWhile: in InitializeInstructionValueSets()
1343 user->opcode() == HloOpcode::kWhile) { in CanShareOperandBufferWithUser()
|
| D | hlo_module_dce.cc | 46 if (instruction->opcode() != HloOpcode::kWhile) { in RunWhileDCE()
|
| D | sharding_propagation.cc | 213 case HloOpcode::kWhile: in PickRepresentativeOperand()
320 case HloOpcode::kWhile: in SupportSpatialPartitioning()
1011 case HloOpcode::kWhile: { in InferShardingFromOperands()
1574 CHECK_EQ(while_instruction->opcode(), HloOpcode::kWhile); in CheckAndUpdateDeviceAssignmentsInWhileBody()
1672 if (inst->opcode() == HloOpcode::kWhile) { in Run()
1711 instruction->opcode() == HloOpcode::kWhile) { in Run()
1726 if (instruction->opcode() == HloOpcode::kWhile) { in Run()
1737 if (instruction->opcode() == HloOpcode::kWhile || in Run()
1758 if (instruction->opcode() == HloOpcode::kWhile) { in Run()
|
| D | flatten_call_graph.cc | 37 case HloOpcode::kWhile: { in ReplaceCalledComputation()
|
| D | while_loop_constant_sinking.cc | 138 return instr->opcode() == HloOpcode::kWhile; in Run()
|
| D | loop_schedule_linearizer.cc | 151 if (instruction->opcode() == HloOpcode::kWhile) { in Run()
|
| D | hlo_opcode.h | 154 V(kWhile, "while", 1)
|
| D | while_loop_analysis.cc | 99 CHECK_EQ(while_op->opcode(), HloOpcode::kWhile); in GetAuxiliaryLoopInductionVars()
242 CHECK_EQ(while_op->opcode(), HloOpcode::kWhile); in GetLoopInductionVarTupleIdx()
|
| D | hlo_element_type_converter.cc | 144 if (opcode == HloOpcode::kWhile || opcode == HloOpcode::kCall || in Run()
|
| D | while_loop_invariant_code_motion.cc | 311 return instr->opcode() == HloOpcode::kWhile; in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | for_thunk.cc | 29 : Thunk(Kind::kWhile, thunk_info), in ForThunk()
|
| D | thunk.cc | 83 case Thunk::kWhile: in ThunkKindToString()
|
| D | while_thunk.cc | 33 : Thunk(Kind::kWhile, thunk_info), in WhileThunk()
|
| D | thunk.h | 71 kWhile, enumerator
|