computeDevice(for:)
Returns the compute device for a compute stage.
Declaration
func computeDevice(for computeStage: ComputeStage) -> MLComputeDevice?Parameters
- computeStage:
The compute stage to inspect.
Return Value
The current compute device; otherwise, nil if one isn’t assigned.