Searched refs:data_init (Results 1 – 2 of 2) sorted by relevance
93 auto data_init = builder.AddInstruction( in BuildWhileInstruction() local99 HloInstruction::CreateTuple({induction_var_init, data_init})) in BuildWhileInstruction()101 HloInstruction::CreateTuple({data_init, induction_var_init})); in BuildWhileInstruction()
641 auto data_init = builder.AddInstruction( in BuildWhileInstruction() local647 HloInstruction::CreateTuple({data_init, data_init})); in BuildWhileInstruction()657 HloInstruction::CreateTuple({induction_var_init, data_init})); in BuildWhileInstruction()666 auto data_init = builder.AddInstruction( in BuildWhileInstruction_InitPointsToConstant() local669 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init, in BuildWhileInstruction_InitPointsToConstant()675 auto data_init = builder.AddInstruction( in BuildWhileInstruction_InitPointsToParameter() local677 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init, in BuildWhileInstruction_InitPointsToParameter()698 auto data_init = builder.AddInstruction(HloInstruction::CreateTernary( in BuildWhileInstruction_InitPointsToAmbiguous() local702 data_init, &builder); in BuildWhileInstruction_InitPointsToAmbiguous()712 auto data_init = in BuildWhileInstruction_InitPointsToNonDistinct() local[all …]