Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_computation_test.cc334 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 …]
Dhlo_computation.cc787 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()
Dcopy_insertion.cc1150 ShapeTree<HloInstruction*> copies_added(indices_to_copy.shape()); in AddSpecialCaseCopies() local
1154 instruction, &indices_to_copy, &copies_added)); in AddSpecialCaseCopies()
Dhlo_computation.h299 ShapeTree<HloInstruction*>* copies_added = nullptr);