license import { Square , SquareInputs } from '@tensorflow/tfjs-core' ; import { KernelConfig } from '@tensorflow/tfjs-core' ; import { MathBackendCPU } from '../backend_cpu' ; import { assertNotComplex } from '../cpu_util' ; export const squareConfig = { kernelName : Square , backendName : 'cpu' , kernelFunc : ( { inputs , backend } ) => { const { x } = inputs as SquareInputs ; const cpuBackend = backend as MathBackendCPU ; assertNotComplex ( x , 'square' ) ; const values = cpuBackend . data . get ( x . dataId ) . values as Float32Array ; const newValues = new Float32Array ( values . length ) ; for ( let i = 0 ; i < values . length ; ++ i ) { const value = values [ i ] ; newValues [ i ] = value * value ; } const dataId = cpuBackend . write ( newValues , x . shape , x . dtype ) ; return { dataId , shape : x . shape , dtype : x . dtype } ; } } ;