Searched refs:TileIndexForDevice (Results 1 – 2 of 2) sorted by relevance
141 std::vector<int64> HloSharding::TileIndexForDevice(int64 device) const { in TileIndexForDevice() function in xla::HloSharding172 std::vector<int64> index = TileIndexForDevice(device); in TileOffsetForDevice()191 std::vector<int64> index = TileIndexForDevice(device); in TileLimitForDevice()
131 std::vector<int64> TileIndexForDevice(int64 device) const;