Searched defs:indices_to_copy (Results 1 – 4 of 4) 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
83 ShapeTree<bool> indices_to_copy(xla_while->shape()); in AddControlEdgesForLoopWrites() local
135 const ShapeTree<bool>& indices_to_copy) { in DeepCopyAndAddControlEdges()175 ShapeTree<bool>* indices_to_copy) { in IndicesToCopyForWhile()203 ShapeTree<bool>* indices_to_copy) { in IndicesToCopyForConditional()289 ShapeTree<bool> indices_to_copy(xla_while->shape()); in AddCopiesForWhile() local1750 ShapeTree<bool> indices_to_copy(conditional->shape()); in AddCopiesForConditional() local1917 const ShapeTree<bool>& indices_to_copy = pair.second; in AddSpecialCaseCopies() local
807 HloInstruction* instruction, const ShapeTree<bool>* indices_to_copy, in DeepCopyInstruction()