license import { backend_util , BroadcastArgs , BroadcastArgsInputs , KernelConfig , TensorInfo , TypedArray } from '@tensorflow/tfjs-core' ; import { MathBackendCPU } from '../backend_cpu' ; export function broadcastArgs ( args ) { const { inputs , backend } = args ; const { s0 , s1 } = inputs ; const s0Vals = backend . data . get ( s0 . dataId ) . values as TypedArray ; const s1Vals = backend . data . get ( s1 . dataId ) . values as TypedArray ; const broadcastShape = backend_util . assertAndGetBroadcastShape ( Array . from ( s0Vals ) , Array . from ( s1Vals ) ) ; return backend . makeTensorInfo ( [ broadcastShape . length ] , 'int32' , Int32Array . from ( broadcastShape ) ) ; } export const broadcastArgsConfig = { kernelName : BroadcastArgs , backendName : 'cpu' , kernelFunc : broadcastArgs } ;