Searched refs:copies_added (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_computation_test.cc | 334 ShapeTree<HloInstruction*> copies_added(tuple->shape(), in TEST_F() local 337 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added) in TEST_F() 342 .Is(copies_added.element({0})), in TEST_F() 344 .Is(copies_added.element({1}))))); in TEST_F() 351 ShapeTree<HloInstruction*> copies_added(tuple->shape(), in TEST_F() local 354 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added) in TEST_F() 360 EXPECT_TRUE(copies_added.element({}) == nullptr); in TEST_F() 361 EXPECT_TRUE(copies_added.element({0}) == nullptr); in TEST_F() 362 EXPECT_TRUE(copies_added.element({1}) == nullptr); in TEST_F() 369 ShapeTree<HloInstruction*> copies_added(tuple->shape(), in TEST_F() local [all …]
|
D | hlo_computation.cc | 787 ShapeTree<HloInstruction*>* copies_added) { in DeepCopyInstruction() argument 803 auto copy_leaf = [indices_to_copy, copies_added]( in DeepCopyInstruction() 809 if (copies_added != nullptr) { in DeepCopyInstruction() 810 *copies_added->mutable_element(leaf_index) = copy; in DeepCopyInstruction()
|
D | copy_insertion.cc | 1150 ShapeTree<HloInstruction*> copies_added(indices_to_copy.shape()); in AddSpecialCaseCopies() local 1154 instruction, &indices_to_copy, &copies_added)); in AddSpecialCaseCopies()
|
D | hlo_computation.h | 299 ShapeTree<HloInstruction*>* copies_added = nullptr);
|