Searched refs:subtree_index (Results 1 – 1 of 1) sorted by relevance
130 xla::ShapeIndex subtree_index = {1}; in TEST() local131 ScopedShapedBuffer output = sb.TakeSubTree(subtree_index); in TEST()135 xla::ShapeIndex orig_index = subtree_index; in TEST()143 if (ShapeIndexView(index).StartsWith(subtree_index)) { in TEST()