Searched refs:TaskAndDevice (Results 1 – 1 of 1) sorted by relevance
219 struct TaskAndDevice { struct220 TaskAndDevice() {} in TaskAndDevice() function221 TaskAndDevice(int task, int device) : task(task), device(device) {} in TaskAndDevice() argument264 StatusOr<xla::Array4D<TaskAndDevice>> ParseTopologyAttr( in ParseTopologyAttr()310 xla::Array4D<TaskAndDevice> topology(bound_x, bound_y, bound_z, bound_core); in ParseTopologyAttr()397 TaskAndDevice task_and_device = topology(x, y, z, core); in GetGeneralTPUExecutionDeviceAssignment()