| /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_util_test.cc | 73 ASSERT_EQ(while_instr->opcode(), HloOpcode::kWhile); in TEST()
104 ASSERT_EQ(while_instr->opcode(), HloOpcode::kWhile); in TEST()
208 return instr->opcode() == HloOpcode::kWhile; in TEST()
|
| 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()
220 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile); in TEST_F()
413 return (instr->opcode() == HloOpcode::kWhile && in TEST_F()
|
| D | bfloat16_propagation.cc | 165 CHECK_EQ(while_hlo->opcode(), HloOpcode::kWhile); in DetermineWhileComputationsPrecision()
254 } else if (use.instruction->opcode() == HloOpcode::kWhile) { in AllUsersConsumeBF16()
322 } else if (hlo->opcode() == HloOpcode::kWhile) { in DetermineInstructionPrecision()
329 if (hlo->opcode() == HloOpcode::kWhile && in DetermineInstructionPrecision()
442 case HloOpcode::kWhile: in AdjustCalledComputationParameters()
493 case HloOpcode::kWhile: in AdjustCalledComputationRoot()
554 if (hlo->opcode() == HloOpcode::kWhile) { in ResolveInconsistencyOfAliasingBuffersHelper()
734 if (inst->opcode() != HloOpcode::kWhile) { in Run()
|
| 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 | 72 if (instr->opcode() == HloOpcode::kWhile) { in TEST()
133 if (instr->opcode() == HloOpcode::kWhile) { in TEST()
|
| D | hlo_ordering.cc | 60 if (a_ancestor == b_ancestor && a_ancestor->opcode() == HloOpcode::kWhile) { in ExecutesBefore()
146 if (b.is_phi() && b.defining_instruction()->opcode() == HloOpcode::kWhile && in IsDefinedBefore()
194 if (use.instruction->opcode() == HloOpcode::kWhile) { in UseIsBeforeValueDefinition()
208 if (value.defining_instruction()->opcode() == HloOpcode::kWhile) { in UseIsBeforeValueDefinition()
|
| 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()
270 if (value.defining_instruction()->opcode() == HloOpcode::kWhile) { in ComputeWhileAliasedBuffers()
|
| D | while_loop_trip_count_annotator.cc | 26 if (instr->opcode() != HloOpcode::kWhile) { in Run()
|
| D | hlo_dataflow_analysis.cc | 490 } else if (callsite.instruction()->opcode() == HloOpcode::kWhile) { in UpdateParameterValueSet()
587 CHECK_EQ(xla_while->opcode(), HloOpcode::kWhile); in UpdateWhileValueSet()
620 case HloOpcode::kWhile: in UpdateInstructionValueSet()
711 } else if (callsite.instruction()->opcode() == HloOpcode::kWhile) { in Propagate()
776 case HloOpcode::kWhile: in InitializeInstructionValueSets()
1096 user->opcode() == HloOpcode::kWhile) { in CanShareOperandBufferWithUser()
|
| D | hlo_module_dce.cc | 43 if (instruction->opcode() != HloOpcode::kWhile) { in RunWhileDCE()
|
| 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 | hlo_opcode.h | 145 V(kWhile, "while", 1)
|
| D | hlo_module_dce_test.cc | 51 if (instruction->opcode() == HloOpcode::kWhile && in WhileBodyHasPassThroughTupleElement()
|
| D | bfloat16_conversion_folding.cc | 164 hlo->opcode() == HloOpcode::kWhile || // in DefaultAction()
|
| D | hlo_instruction.cc | 944 absl::WrapUnique(new HloInstruction(HloOpcode::kWhile, shape)); in CreateWhile()
1472 case HloOpcode::kWhile: in CloneWithNewOperands()
1793 case HloOpcode::kWhile: in IdenticalSlowPath()
2027 CHECK_EQ(HloOpcode::kWhile, opcode_); in while_condition()
2032 CHECK_EQ(HloOpcode::kWhile, opcode_); in while_body()
2040 CHECK_EQ(HloOpcode::kWhile, opcode_); in set_while_condition()
2048 CHECK_EQ(HloOpcode::kWhile, opcode_); in set_while_body()
2053 CHECK_EQ(HloOpcode::kWhile, opcode_); in while_init()
2281 if (opcode() == HloOpcode::kWhile) { in ExtraAttributesToString()
2327 case HloOpcode::kWhile: in ExtraAttributesToString()
[all …]
|
| D | hlo_module_group_metadata.cc | 301 if (hlo->opcode() == HloOpcode::kWhile) { in RecordInstructions()
384 TF_RET_CHECK(instruction1->opcode() == HloOpcode::kWhile || in AddCompanion()
|
| D | while_loop_invariant_code_motion.cc | 310 return instr->opcode() == HloOpcode::kWhile; in Run()
|
| D | hlo_element_type_converter.cc | 144 if (opcode == HloOpcode::kWhile || opcode == HloOpcode::kCall || in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk.cc | 59 case Thunk::kWhile: in ThunkKindToString()
|
| D | for_thunk.cc | 29 : Thunk(Kind::kWhile, hlo), in ForThunk()
|
| D | thunk.h | 63 kWhile, enumerator
|
| D | while_thunk.cc | 31 : Thunk(Kind::kWhile, hlo), in WhileThunk()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.cc | 192 if (instruction->opcode() == HloOpcode::kWhile) { in AssignParallelTasksHelper()
|