Searched defs:dtensor_device_name (Results 1 – 1 of 1) sorted by relevance
94 const MeshWithParallelDevice& mesh, const std::string& dtensor_device_name, in BroadcastResourceTensor()274 const MeshWithParallelDevice& mesh, const std::string& dtensor_device_name, in Broadcast()