license import { KernelConfig , KernelFunc , Sum , SumAttrs , SumInputs } from '@tensorflow/tfjs-core' ; import { MathBackendWebGL } from '../backend_webgl' ; import { sumImpl } from './Sum_impl' ; export function sum ( args ) { const { inputs , backend , attrs } = args ; const { x } = inputs ; const { axis , keepDims } = attrs ; return sumImpl ( x , axis , keepDims , backend ) ; } export const sumConfig = { kernelName : Sum , backendName : 'webgl' , kernelFunc : sum as { } as KernelFunc } ;