Searched refs:BuildWhileInstructionWithCustomInit (Results 1 – 1 of 1) sorted by relevance
604 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init, in BuildWhileInstruction_InitPointsToConstant()612 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init, in BuildWhileInstruction_InitPointsToParameter()626 return BuildWhileInstructionWithCustomInit(nested_loop_state_shape_, in BuildWhileInstruction_InitPointsToNonDistinct()643 auto xla_while = BuildWhileInstructionWithCustomInit(loop_state_shape_, in BuildWhileInstruction_InitPointsToInterfering()664 HloInstruction* BuildWhileInstructionWithCustomInit( in BuildWhileInstructionWithCustomInit() function in xla::__anon2db0cb110111::WhileCopyInsertionTest