Searched defs:SetTPUCoreIDs (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ | ||
| H A D | dtensor_device.cc | 161 Status SetTPUCoreIDs(const std::string& mesh_name, in SetTPUCoreIDs() function in tensorflow::dtensor::DTensorDevice |