Searched defs:indices_to_copy (Results 1 – 4 of 4) sorted by relevance
286 ShapeTree<bool> indices_to_copy(constant->shape(), /*init_value=*/true); in TEST_F() local294 ShapeTree<bool> indices_to_copy(constant->shape(), /*init_value=*/false); in TEST_F() local315 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/true); in TEST_F() local332 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/false); in TEST_F() local349 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/false); in TEST_F() local
139 const ShapeTree<bool>& indices_to_copy) { in DeepCopyAndAddControlEdges()179 ShapeTree<bool>* indices_to_copy) { in IndicesToCopyForWhile()207 ShapeTree<bool>* indices_to_copy) { in IndicesToCopyForConditional()293 ShapeTree<bool> indices_to_copy(xla_while->shape()); in AddCopiesForWhile() local1787 ShapeTree<bool> indices_to_copy(conditional->shape()); in AddCopiesForConditional() local1993 const ShapeTree<bool>& indices_to_copy = pair.second; in AddSpecialCaseCopies() local
83 ShapeTree<bool> indices_to_copy(xla_while->shape()); in AddControlEdgesForLoopWrites() local
829 HloInstruction* instruction, const ShapeTree<bool>* indices_to_copy, in DeepCopyInstruction()