Searched defs:tpu_device (Results 1 – 7 of 7) sorted by relevance
76 string tpu_device; in ComputeAsync() local233 std::string tpu_device; in Compute() local
473 def generate_dequeue_op(self, tpu_device=0): argument787 def generate_dequeue_op(self, tpu_device=0): argument
170 def tpu_device(self, member in DeviceAssignment
176 std::string GetCPUHostDeviceForTPUDevice(Device tpu_device) { in GetCPUHostDeviceForTPUDevice()208 const auto& tpu_device = tpu_devices[task][device]; in GetFullMeshTPUExecutionDeviceAssignment() local413 const auto& tpu_device = tpu_devices[task][device]; in GetGeneralTPUExecutionDeviceAssignment() local
73 llvm::StringRef tpu_device = tpu_replica_0.cast<ArrayAttr>()[replica_id] in GetDeviceOrdinal() local
92 DeviceProperties tpu_device; in TEST() local
3464 const DeviceNameUtils::ParsedName& tpu_device, in CopyOutsideCompilationNodes()3524 DeviceNameUtils::ParsedName tpu_device; in ReplicateOutsideCompilationNodes() local