WASM backend error - Error: 'complex' not yet implemented or not found in the registry

[shilsircar]

To get help from the community, we encourage using Stack Overflow and the tensorflow.js tag.

TensorFlow.js version

2.0.1

Browser version

Chrome m82

Describe the problem or feature request

I get this error trying the wasm backend on a custom model:

Uncaught (in promise) Error: 'complex' not yet implemented or not found in the registry. Did you forget to import the kernel? at ZN (tfjs:17) at BackendWasm.e.complex (tfjs:17) at tfjs:17 at tfjs:17 at tfjs:17 at t.e.scopedRun (tfjs:17) at t.e.tidy (tfjs:17) at h (tfjs:17) at tfjs:17 at t.e.scopedRun (tfjs:17) Uncaught (in promise) Error: 'complex' not yet implemented or not found in the registry. Did you forget to import the kernel?

is complex implemented for wasm backend ?

Code to reproduce the bug / link to feature request

If you would like to get help from the community, we encourage using Stack Overflow and the tensorflow.js tag.

GitHub issues for this repository are tracked in the tfjs union repository.

Please file your issue there, following the guidance in that issue template.

[annxingyuan]

@shilsircar thanks for filing this - WASM does not yet have support for complex ops. I have a PR in progress to add complex support to WASM. Will keep you posted!

[rthadur]

Here is the related draft from Ann https://github.com/tensorflow/tfjs/pull/3478/files

[Ahmed-Z]

Is there any progress for adding complex support to WASM ?

[gaikwadrahul8]

Hi, @Ahmed-Z

Apologize for the delayed response and at the moment TFJs WASM backend does not support complex dtype and all kernels/ops for complex tensors and team will work on it later this year and you can refer this comment . Thank you!