Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dcopy_insertion.cc130 const ShapeTree<bool>& indices_to_copy) { in DeepCopyAndAddControlEdges() argument
139 from, &indices_to_copy, &from_copy_tree)); in DeepCopyAndAddControlEdges()
144 to->parent()->DeepCopyInstruction(to, &indices_to_copy, &to_copy_tree)); in DeepCopyAndAddControlEdges()
170 ShapeTree<bool>* indices_to_copy) { in IndicesToCopyForWhile() argument
171 DCHECK(ShapeUtil::Compatible(indices_to_copy->shape(), xla_while->shape())); in IndicesToCopyForWhile()
175 for (auto& pair : *indices_to_copy) { in IndicesToCopyForWhile()
261 ShapeTree<bool> indices_to_copy(xla_while->shape()); in AddCopiesForWhile() local
263 &indices_to_copy)) { in AddCopiesForWhile()
270 for (auto& pair : indices_to_copy) { in AddCopiesForWhile()
280 xla_while->parent()->DeepCopyInstruction(while_init, &indices_to_copy)); in AddCopiesForWhile()
[all …]
Dhlo_computation_test.cc305 ShapeTree<bool> indices_to_copy(constant->shape(), /*init_value=*/true); in TEST_F() local
306 EXPECT_THAT(computation->DeepCopyInstruction(constant, &indices_to_copy) in TEST_F()
313 ShapeTree<bool> indices_to_copy(constant->shape(), /*init_value=*/false); in TEST_F() local
314 EXPECT_EQ(computation->DeepCopyInstruction(constant, &indices_to_copy) in TEST_F()
334 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/true); in TEST_F() local
338 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added) in TEST_F()
351 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/false); in TEST_F() local
355 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added) in TEST_F()
368 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/false); in TEST_F() local
369 *indices_to_copy.mutable_element({0}) = true; in TEST_F()
[all …]
Dhlo_computation.cc658 HloInstruction* instruction, const ShapeTree<bool>* indices_to_copy, in DeepCopyInstruction() argument
665 if (indices_to_copy != nullptr && in DeepCopyInstruction()
666 !ShapeUtil::Compatible(instruction->shape(), indices_to_copy->shape())) { in DeepCopyInstruction()
671 ShapeUtil::HumanString(indices_to_copy->shape())); in DeepCopyInstruction()
675 auto copy_leaf = [indices_to_copy, copies_added]( in DeepCopyInstruction()
678 if (indices_to_copy == nullptr || indices_to_copy->element(leaf_index)) { in DeepCopyInstruction()
Dhlo_computation.h257 const ShapeTree<bool>* indices_to_copy = nullptr,