Home
last modified time | relevance | path

Searched refs:BuildWhileInstructionWithCustomInit (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcopy_insertion_test.cc604 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