Lines Matching defs:subshape
233 const Shape& subshape = shape.tuple_shapes(i); in mutable_shape_do_not_use() local
251 for (const Shape& subshape : shape.tuple_shapes()) { in SetPiece() local
676 Shape* subshape = in SetDynamicSize() local
853 Shape* subshape = ShapeUtil::GetMutableSubshape(&new_shape, shape_index); in Relayout() local
869 [this, &result](const Shape& subshape, const ShapeIndex& index) { in Relayout()
883 shape(), [&](const Shape& subshape, const ShapeIndex& index) { in ToBoundedDynamic()
900 &new_shape, [this](Shape* subshape, const ShapeIndex& index) { in ToStatic()
1128 const Shape& subshape = ShapeUtil::GetSubshape(shape(), shape_index); in GetAsString() local
1304 const Shape& subshape = ShapeUtil::GetSubshape(literal.shape(), shape_index); in TupleToStringHelper() local
1325 const Shape& subshape = ShapeUtil::GetSubshape(literal.shape(), shape_index); in DenseArrayToStringHelper() local
1413 const Shape& subshape = ShapeUtil::GetSubshape(literal.shape(), shape_index); in ToStringHelper() local
1882 const Shape& subshape = piece.subshape(); in operator ==() local
2566 const Shape& subshape = shape.tuple_shapes(i); in CopyPieceSubtree() local
2688 const Shape& subshape = shape.tuple_shapes(i); in BuildPieceSubtree() local