Searched refs:copies_added (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_computation_test.cc | 335 ShapeTree<HloInstruction*> copies_added(tuple->shape(), in TEST_F() local
338 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added) in TEST_F()
343 .Is(copies_added.element({0})), in TEST_F()
345 .Is(copies_added.element({1}))))); in TEST_F()
352 ShapeTree<HloInstruction*> copies_added(tuple->shape(), in TEST_F() local
355 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added) in TEST_F()
361 EXPECT_TRUE(copies_added.element({}) == nullptr); in TEST_F()
362 EXPECT_TRUE(copies_added.element({0}) == nullptr); in TEST_F()
363 EXPECT_TRUE(copies_added.element({1}) == nullptr); in TEST_F()
370 ShapeTree<HloInstruction*> copies_added(tuple->shape(), in TEST_F() local
[all …]
|
| D | hlo_computation.cc | 659 ShapeTree<HloInstruction*>* copies_added) { in DeepCopyInstruction() argument
675 auto copy_leaf = [indices_to_copy, copies_added]( in DeepCopyInstruction()
681 if (copies_added != nullptr) { in DeepCopyInstruction()
682 *copies_added->mutable_element(leaf_index) = copy; in DeepCopyInstruction()
|
| D | hlo_computation.h | 258 ShapeTree<HloInstruction*>* copies_added = nullptr);
|
| D | copy_insertion.cc | 1076 ShapeTree<HloInstruction*> copies_added(indices_to_copy.shape()); in AddSpecialCaseCopies() local
1080 instruction, &indices_to_copy, &copies_added)); in AddSpecialCaseCopies()
|