license import { FFT , FFTInputs , KernelConfig } from '@tensorflow/tfjs' ; import { createTensorsTypeOpAttr , NodeJSKernelBackend } from '../nodejs_kernel_backend' ; export const FFTConfig = { kernelName : FFT , backendName : 'tensorflow' , kernelFunc : ( args ) => { const { input } = args . inputs as FFTInputs ; const backend = args . backend as NodeJSKernelBackend ; const opAttrs = [ createTensorsTypeOpAttr ( 'Tcomplex' , input . dtype ) ] ; return backend . executeSingleOutput ( FFT , opAttrs , [ input ] ) ; } } ;