Searched refs:ExtractPrefix (Results 1 – 5 of 5) sorted by relevance
30 static HloInstruction* ExtractPrefix(HloInstruction* input_tuple,47 return ExtractPrefix(input_tuple, input_tuple->shape().tuple_shapes_size()); in Duplicate()
58 TEST(TupleUtilTest, ExtractPrefix) { in TEST() argument65 HloInstruction* prefix = TupleUtil::ExtractPrefix(param0, 2); in TEST()
50 TupleUtil::ExtractPrefix(wide_while_cond->parameter_instruction(0), in WidenWhileCondition()74 HloInstruction* truncated_parameter = TupleUtil::ExtractPrefix( in WidenWhileBody()128 HloInstruction* replacement_instr = TupleUtil::ExtractPrefix( in MakeInstructionsLiveIn()
22 /*static*/ HloInstruction* TupleUtil::ExtractPrefix(HloInstruction* input_tuple, in ExtractPrefix() function in xla::TupleUtil
57 HloInstruction* truncated_parameter = TupleUtil::ExtractPrefix( in WidenComputation()1377 HloInstruction* new_conditional_extracted = TupleUtil::ExtractPrefix( in HandleConditional()