W tensorflow/core/framework/op_kernel.cc:1780] OP_REQUIRES failed at xla_ops.cc:296 : UNIMPLEMENTED: Could not find compiler for platform DML: NOT_FOUND: could not find registered compiler for platform DML -- was support for that platform linked in?
DML doesn't support XLA because it is fundamentally different from how the compiler for other backends like CUDA and AMD works. To make the code work, remove the jit_compile=True line above.
It seems do not support XLA (jit compile), such as the following demo:
And the bug is shown as:
Thanks in advance. Best wishes.