Searched defs:tile_shape (Results 1 – 2 of 2) sorted by relevance
35 OpSharding Tile(const Shape& tile_shape, in Tile()49 OpSharding Tile1D(const Shape& tile_shape, int64 num_tiles) { in Tile1D()
589 ShapeProto tile_shape = 2; field