license import { KernelConfig , KernelFunc , Range , RangeAttrs , TensorInfo } from '@tensorflow/tfjs-core' ; import { MathBackendCPU } from '../backend_cpu' ; import { rangeImpl } from './Range_impl' ; export function range ( args ) { const { backend , attrs } = args ; const { start , stop , dtype , step } = attrs ; const values = rangeImpl ( start , stop , step , dtype ) ; return backend . makeTensorInfo ( [ values . length ] , dtype , values ) ; } export const rangeConfig = { kernelName : Range , backendName : 'cpu' , kernelFunc : range as { } as KernelFunc } ;