Searched refs:tuple_copy (Results 1 – 2 of 2) sorted by relevance
286 auto tuple_copy = computation->DeepCopyInstruction(tuple).ValueOrDie(); in TEST_F() local288 EXPECT_THAT(tuple_copy, GmockMatch(m::Tuple( in TEST_F()291 EXPECT_EQ(0, tuple_copy->operand(0)->operand(0)->tuple_index()); in TEST_F()292 EXPECT_EQ(1, tuple_copy->operand(1)->operand(0)->tuple_index()); in TEST_F()
804 HloInstruction* tuple_copy = instruction->parent()->AddInstruction( in CreateCopyWithNewLayout() local806 SetupCopiedInstruction(*instruction, tuple_copy, {}); in CreateCopyWithNewLayout()807 LayoutUtil::ClearLayout(tuple_copy->mutable_shape()); in CreateCopyWithNewLayout()809 shape_with_layout, tuple_copy->mutable_shape())); in CreateCopyWithNewLayout()810 return tuple_copy; in CreateCopyWithNewLayout()