Searched refs:BuildWhileInstructionWithCustomInit (Results 1 – 1 of 1) sorted by relevance
669 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init, in BuildWhileInstruction_InitPointsToConstant()677 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init, in BuildWhileInstruction_InitPointsToParameter()701 return BuildWhileInstructionWithCustomInit(nested_loop_state_shape_, in BuildWhileInstruction_InitPointsToAmbiguous()715 return BuildWhileInstructionWithCustomInit(nested_loop_state_shape_, in BuildWhileInstruction_InitPointsToNonDistinct()732 auto xla_while = BuildWhileInstructionWithCustomInit(loop_state_shape_, in BuildWhileInstruction_InitPointsToInterfering()753 HloInstruction* BuildWhileInstructionWithCustomInit( in BuildWhileInstructionWithCustomInit() function in xla::__anon1fa5445b0111::WhileCopyInsertionTest