Searched refs:TileIndexForDevice (Results 1 – 2 of 2) sorted by relevance
230 std::vector<int64> HloSharding::TileIndexForDevice(int64 device) const { in TileIndexForDevice() function in xla::HloSharding276 std::vector<int64> index = TileIndexForDevice(device); in TileOffsetForDevice()297 std::vector<int64> index = TileIndexForDevice(device); in TileLimitForDevice()584 std::vector<int64> index = TileIndexForDevice(device); in TileShape()
171 std::vector<int64> TileIndexForDevice(int64 device) const;