Searched defs:tpu_core_locations (Results 1 – 3 of 3) sorted by relevance
355 def tpu_core_locations_to_ids(self, tpu_core_locations): argument
201 const std::vector<std::vector<int>>& tpu_core_locations) { in PYBIND11_MODULE()
188 std::vector<std::vector<int>> tpu_core_locations; in TPUCoreIDsToLocations() local218 const std::vector<std::vector<int>>& tpu_core_locations) { in TPUCoreLocationsToIDs()2112 const std::vector<std::vector<int>>& tpu_core_locations, in TPUCoreLocationsToIDs()