Searched defs:indices_to_copy (Results 1 – 3 of 3) sorted by relevance
130 const ShapeTree<bool>& indices_to_copy) { in DeepCopyAndAddControlEdges()170 ShapeTree<bool>* indices_to_copy) { in IndicesToCopyForWhile()261 ShapeTree<bool> indices_to_copy(xla_while->shape()); in AddCopiesForWhile() local1074 const ShapeTree<bool>& indices_to_copy = pair.second; in AddSpecialCaseCopies() local1257 ShapeTree<bool> indices_to_copy(root->shape(), /*init_value=*/false); in AddCopiesForBufferAssignment() local
305 ShapeTree<bool> indices_to_copy(constant->shape(), /*init_value=*/true); in TEST_F() local313 ShapeTree<bool> indices_to_copy(constant->shape(), /*init_value=*/false); in TEST_F() local334 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/true); in TEST_F() local351 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/false); in TEST_F() local368 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/false); in TEST_F() local
658 HloInstruction* instruction, const ShapeTree<bool>* indices_to_copy, in DeepCopyInstruction()