Post

Replies

Boosts

Views

Activity

Comment on Cannot assign a device for operation ReadVariableOp
Is there any update on when tensorflow-metal will implement these missing operations? I am using: tensorflow-macos 2.8.0 tensorflow-metal 0.4.0 I am getting the following error which I assume is the same issue as above. Colocation Debug Info: Colocation group had the following types and supported devices: Root Member(assigned_device_name_index_=2 requested_device_name_='/job:localhost/replica:0/task:0/device:GPU:0' assigned_device_name_='/job:localhost/replica:0/task:0/device:GPU:0' resource_device_name_='/job:localhost/replica:0/task:0/device:GPU:0' supported_device_types_=[CPU] possible_devices_=[] ReadVariableOp: GPU CPU ResourceApplyAdam: CPU _Arg: GPU CPU Colocation members, user-requested devices, and framework assigned devices, if any: sequential_dense_matmul_readvariableop_resource (_Arg) framework assigned device=/job:localhost/replica:0/task:0/device:GPU:0 adam_adam_update_resourceapplyadam_m (_Arg) framework assigned device=/job:localhost/replica:0/task:0/device:GPU:0 adam_adam_update_resourceapplyadam_v (_Arg) framework assigned device=/job:localhost/replica:0/task:0/device:GPU:0 sequential/dense/MatMul/ReadVariableOp (ReadVariableOp) Adam/Adam/update/ResourceApplyAdam (ResourceApplyAdam) /job:localhost/replica:0/task:0/device:GPU:0 [[{{node sequential/dense/MatMul/ReadVariableOp}}]] [Op:__inference_train_function_809]
Apr ’22