Searched defs:DTensorDevice (Results 1 – 3 of 3) sorted by relevance
558 def _set_dtensor_device(device: dtensor_device.DTensorDevice) -> None:
48 class DTensorDevice(object): class
93 class DTensorDevice { class95 explicit DTensorDevice(absl::string_view name) in DTensorDevice() function in tensorflow::dtensor::DTensorDevice