Home
last modified time | relevance | path

Searched refs:while_condition (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_constant_sinking_test.cc280 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F() local
281 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F()
320 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F() local
321 EXPECT_THAT(while_condition->root_instruction(), in TEST_F()
363 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F() local
364 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F()
365 for (const HloInstruction* inst : while_condition->instructions()) { in TEST_F()
413 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F() local
414 EXPECT_THAT(while_condition->root_instruction(), in TEST_F()
Dhlo_rematerialization_test_utils_test.cc70 auto while_condition = MakeConditionComputation(); in TEST_F() local
73 while_condition.get(), body_computation.get()); in TEST_F()
92 EXPECT_EQ(instructions[4]->while_condition()->name(), in TEST_F()
Dwhile_loop_simplifier_test.cc234 auto* while_cond = while_op->while_condition(); in TEST_F()
442 EXPECT_THAT(new_while_op->while_condition()->root_instruction(), in TEST_F()
619 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
704 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
774 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
782 EXPECT_THAT(new_while->while_condition()->root_instruction(), in TEST_F()
839 new_while->while_condition()->parameter_instruction(0)->shape(), in TEST_F()
Dhlo_ordering.cc81 const HloComputation* condition = a_ancestor->while_condition(); in GetExecutionConstraint()
193 b.defining_instruction()->while_condition()))) { in IsDefinedBefore()
280 xla_while->while_condition())) { in UsesBeforeValueDefinition()
282 xla_while->while_condition()->parameter_instruction(0)) { in UsesBeforeValueDefinition()
301 xla_while->while_condition())) { in UsesBeforeValueDefinition()
Dwhile_loop_simplifier.cc53 HloComputation* while_cond = while_op->while_condition(); in RemoveDeadTupleIndices()
225 HloComputation* while_cond = while_op->while_condition(); in TryRemoveDeadWhileParams()
337 HloComputation* while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndicesHelper()
399 HloComputation* while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndices()
479 while_cond = while_op->while_condition(); in TryRemoveRepeatedWhileTupleIndices()
496 auto* while_cond = while_op->while_condition(); in TryRemoveConstantParams()
651 if (while_op->while_condition()->HasSideEffect()) { in TryRemoveWhileLoop()
782 propagate_constant(while_op->while_condition())); in TryPropagateConstant()
859 auto* while_cond = while_op->while_condition(); in TryFlattenNestedTuples()
986 auto* while_cond = while_op->while_condition(); in TryMergeInductionVariables()
[all …]
Dhlo_replication_analysis.cc210 inst->while_condition()->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
213 inst->while_condition()->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
221 inst->while_condition(), mark_everything_not_replicated); in ComputeHloReplicationOnComputation()
225 !hlo_replication_[inst->while_condition()->root_instruction()] in ComputeHloReplicationOnComputation()
Dwhile_loop_analysis.cc252 auto* while_cond = while_op->while_condition(); in GetLoopInductionVarTupleIdx()
406 auto* while_cond = while_op->while_condition(); in PatternMatchLoopTripCount()
511 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCount()
580 auto* while_cond = while_op->while_condition(); in ComputeWhileLoopTripCountUpperBound()
Dbfloat16_propagation.cc176 HloComputation* condition = while_hlo->while_condition(); in DetermineWhileComputationsPrecision()
285 use.instruction->while_condition()->parameter_instruction( in AllUsersConsumeBF16()
391 (caller_counts_[hlo->while_condition()] > 1 || in DetermineInstructionPrecision()
506 adjust_computation(hlo->while_condition(), hlo->operands()); in AdjustCalledComputationParameters()
674 hlo->while_condition(), &visited_in_while) || in ResolveInconsistencyOfAliasingBuffersHelper()
Dhlo_liveness_analysis.cc186 MarkLiveAtIndex(instruction->while_condition()->root_instruction(), {}, in PropagateLivenessThroughWhile()
238 MarkLiveAtIndex(caller->while_condition()->root_instruction(), {}, in PropagateLivenessThroughControlFlow()
Dflatten_call_graph.cc38 if (computation == instruction->while_condition()) { in ReplaceCalledComputation()
Dwhile_loop_constant_sinking.cc51 HloComputation* while_cond = while_instr->while_condition(); in TrySinkingConstantsIntoWhileLoop()
Dhlo_live_range.cc112 time = FlattenSchedule(*instruction->while_condition(), time); in FlattenSchedule()
Dhlo_schedule_test.cc254 HloComputation* cond = xla_while->while_condition(); in TEST_F()
Dcall_inliner_test.cc113 computation->root_instruction()->while_condition()->root_instruction(), in TEST_F()
Dcopy_insertion_test.cc862 module_->AddEmbeddedComputation(while_hlo->while_condition()->Clone()); in TEST_F()
880 while_hlo->shape(), while_hlo->while_condition(), in TEST_F()
930 module_->AddEmbeddedComputation(while_hlo->while_condition()->Clone()); in TEST_F()
940 while_hlo->shape(), while_hlo->while_condition(), in TEST_F()
990 module_->AddEmbeddedComputation(while_hlo->while_condition()->Clone()); in TEST_F()
1008 while_hlo->shape(), while_hlo->while_condition(), in TEST_F()
Dwhile_util.cc110 WidenWhileCondition(while_instr->while_condition(), new_while_shape)); in MakeInstructionsLiveIn()
Dhlo_verifier.cc924 CheckParameterCount(xla_while, xla_while->while_condition(), 1)); in HandleWhile()
928 CheckOperandAndParameter(xla_while, 0, xla_while->while_condition(), 0)); in HandleWhile()
930 xla_while->while_condition()->root_instruction()->shape(); in HandleWhile()
1816 auto* while_cond = xla_while->while_condition(); in HandleWhile()
Dwhile_loop_all_reduce_code_motion.cc502 while_instruction->while_condition(), while_instruction->while_body(), in AddSinkedAllReducesAndReplaceWhile()
Dhlo_module_group_metadata.cc319 tracked_instructions_[hlo->while_condition()] = in RecordInstructions()
Dhlo_dataflow_analysis.cc64 ordinal = CalculatePostOrderScheduleHelper(instruction->while_condition(), in CalculatePostOrderScheduleHelper()
916 callsite.instruction()->while_condition()->parameter_instruction( in Propagate()
Dhlo_instruction.h1291 HloComputation* while_condition() const;
1293 void set_while_condition(HloComputation* while_condition);
Dhlo_instruction.cc1795 CreateWhile(shape, while_condition(), while_body(), new_operands[0]); in CloneWithNewOperands()
2195 eq_computations(while_condition(), other.while_condition())); in IdenticalSlowPath()
2468 HloComputation* HloInstruction::while_condition() const { in while_condition() function in xla::HloInstruction
2796 "condition=", PrintNameInternal(while_condition()->name(), options))); in ExtraAttributesToString()
2857 StrCat("condition=\n", while_condition()->ToString(new_options))); in ExtraAttributesToString()
Dhlo_module.cc184 replacements, instruction->while_condition(), nullptr); in ReplaceComputations()
/external/tensorflow/tensorflow/python/keras/
Dconstraints.py254 while_condition = lambda index, *args: K.less(index, start) function
263 while_condition,
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2359 HloComputation* condition = xla_while->while_condition(); in HandleWhile()
2386 xla_while, xla_while->while_condition()->parameter_instruction(0), in HandleWhile()
2416 EmitGlobalCall(*xla_while->while_condition(), IrName(xla_while, "cond")); in HandleWhile()
2418 Load(GetBufferForGlobalCallReturnValue(*xla_while->while_condition())), in HandleWhile()

12