license import { Imag , ImagInputs , KernelConfig , KernelFunc , TensorInfo } from '@tensorflow/tfjs-core' ; import { MathBackendCPU } from '../backend_cpu' ; export function imag ( args ) { const { inputs , backend } = args ; const { input } = inputs ; const imag = backend . data . get ( input . dataId ) . complexTensorInfos . imag ; const imagVal = backend . data . get ( imag . dataId ) . values ; return backend . makeTensorInfo ( imag . shape , imag . dtype , imagVal ) ; } export const imagConfig = { kernelName : Imag , backendName : 'cpu' , kernelFunc : imag as { } as KernelFunc } ;