license import { KernelConfig , Square } from '@tensorflow/tfjs-core' ; import { unaryKernelFunc } from '../kernel_utils/kernel_funcs_utils' ; const SQUARE = `return x * x;` ; export const square = unaryKernelFunc ( { opSnippet : SQUARE } ) ; export const squareConfig = { kernelName : Square , backendName : 'webgl' , kernelFunc : square , } ;