Searched defs:indices_to_copy (Results 1 – 3 of 3) sorted by relevance
304 ShapeTree<bool> indices_to_copy(constant->shape(), /*init_value=*/true); in TEST_F() local312 ShapeTree<bool> indices_to_copy(constant->shape(), /*init_value=*/false); in TEST_F() local333 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/true); in TEST_F() local350 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/false); in TEST_F() local367 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/false); in TEST_F() local
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() local1071 const ShapeTree<bool>& indices_to_copy = pair.second; in AddSpecialCaseCopies() local
766 HloInstruction* instruction, const ShapeTree<bool>* indices_to_copy, in DeepCopyInstruction()