license import { backend_util , BroadcastArgs , BroadcastArgsInputs , KernelConfig } from '@tensorflow/tfjs' ; import { createTensorsTypeOpAttr , NodeJSKernelBackend } from '../nodejs_kernel_backend' ; export const broadcastArgsConfig = { kernelName : BroadcastArgs , backendName : 'tensorflow' , kernelFunc : ( args ) => { const { s0 , s1 } = args . inputs as BroadcastArgsInputs ; const backend = args . backend as NodeJSKernelBackend ; const opAttrs = [ createTensorsTypeOpAttr ( 'T' , backend_util . upcastType ( s0 . dtype , s1 . dtype ) ) ] ; return backend . executeSingleOutput ( BroadcastArgs , opAttrs , [ s0 , s1 ] ) ; } } ;