license import { KernelConfig , KernelFunc , LinSpace , LinSpaceAttrs , TensorInfo } from '@tensorflow/tfjs-core' ; import { MathBackendCPU } from '../backend_cpu' ; import { linSpaceImpl } from './LinSpace_impl' ; export function linSpace ( args ) { const { backend , attrs } = args ; const { start , stop , num } = attrs ; const outVals = linSpaceImpl ( start , stop , num ) ; return backend . makeTensorInfo ( [ outVals . length ] , 'float32' , outVals ) ; } export const linSpaceConfig = { kernelName : LinSpace , backendName : 'cpu' , kernelFunc : linSpace as { } as KernelFunc } ;