Searched refs:parallel_device_mesh (Results 1 – 1 of 1) sorted by relevance
352 const MeshWithParallelDevice* parallel_device_mesh,1457 const MeshWithParallelDevice* parallel_device_mesh, in ExecuteFunctionAndWait() argument1462 parallel_device_mesh->parallel_device().StartExecute( in ExecuteFunctionAndWait()1472 parallel_device_mesh->parallel_device().Join( in ExecuteFunctionAndWait()1546 const MeshWithParallelDevice* parallel_device_mesh = in ExecuteRegularOperation() local1549 if (parallel_device_mesh == nullptr) { in ExecuteRegularOperation()1554 parallel_device_mesh; in ExecuteRegularOperation()1657 const MeshWithParallelDevice* parallel_device_mesh = in ExecuteRegularOperation() local1677 parallel_device_mesh->DeviceIDs(context, status)); in ExecuteRegularOperation()1685 parallel_device_mesh->parallel_device().StartExecute( in ExecuteRegularOperation()[all …]