license import { KernelConfig , registerKernel } from '@tensorflow/tfjs-core' ; import { _fusedMatMulConfig } from './kernels/_FusedMatMul' ; import { absConfig } from './kernels/Abs' ; import { acosConfig } from './kernels/Acos' ; import { acoshConfig } from './kernels/Acosh' ; import { addConfig } from './kernels/Add' ; import { addNConfig } from './kernels/AddN' ; import { allConfig } from './kernels/All' ; import { anyConfig } from './kernels/Any' ; import { argMaxConfig } from './kernels/ArgMax' ; import { argMinConfig } from './kernels/ArgMin' ; import { asinConfig } from './kernels/Asin' ; import { asinhConfig } from './kernels/Asinh' ; import { atanConfig } from './kernels/Atan' ; import { atan2Config } from './kernels/Atan2' ; import { atanhConfig } from './kernels/Atanh' ; import { avgPoolConfig } from './kernels/AvgPool' ; import { avgPool3DConfig } from './kernels/AvgPool3D' ; import { avgPool3DGradConfig } from './kernels/AvgPool3DGrad' ; import { avgPoolGradConfig } from './kernels/AvgPoolGrad' ; import { batchMatMulConfig } from './kernels/BatchMatMul' ; import { batchNormConfig } from './kernels/BatchNorm' ; import { batchToSpaceNDConfig } from './kernels/BatchToSpaceND' ; import { bincountConfig } from './kernels/Bincount' ; import { broadcastArgsConfig } from './kernels/BroadcastArgs' ; import { castConfig } from './kernels/Cast' ; import { ceilConfig } from './kernels/Ceil' ; import { clipByValueConfig } from './kernels/ClipByValue' ; import { complexConfig } from './kernels/Complex' ; import { complexAbsConfig } from './kernels/ComplexAbs' ; import { concatConfig } from './kernels/Concat' ; import { conv2DConfig } from './kernels/Conv2D' ; import { conv2DBackpropFilterConfig } from './kernels/Conv2DBackpropFilter' ; import { conv2DBackpropInputConfig } from './kernels/Conv2DBackpropInput' ; import { conv3DConfig } from './kernels/Conv3D' ; import { conv3DBackpropFilterV2Config } from './kernels/Conv3DBackpropFilterV2' ; import { conv3DBackpropInputConfig } from './kernels/Conv3DBackpropInputV2' ; import { cosConfig } from './kernels/Cos' ; import { coshConfig } from './kernels/Cosh' ; import { cropAndResizeConfig } from './kernels/CropAndResize' ; import { cumsumConfig } from './kernels/Cumsum' ; import { denseBincountConfig } from './kernels/DenseBincount' ; import { depthToSpaceConfig } from './kernels/DepthToSpace' ; import { depthwiseConv2dNativeConfig } from './kernels/DepthwiseConv2dNative' ; import { depthwiseConv2dNativeBackpropFilterConfig } from './kernels/DepthwiseConv2dNativeBackpropFilter' ; import { depthwiseConv2dNativeBackpropInputConfig } from './kernels/DepthwiseConv2dNativeBackpropInput' ; import { diagConfig } from './kernels/Diag' ; import { dilation2DConfig } from './kernels/Dilation2D' ; import { einsumConfig } from './kernels/Einsum' ; import { eluConfig } from './kernels/Elu' ; import { eluGradConfig } from './kernels/EluGrad' ; import { equalConfig } from './kernels/Equal' ; import { erfConfig } from './kernels/Erf' ; import { expConfig } from './kernels/Exp' ; import { expandDimsConfig } from './kernels/ExpandDims' ; import { expm1Config } from './kernels/Expm1' ; import { fftConfig } from './kernels/FFT' ; import { fillConfig } from './kernels/Fill' ; import { flipLeftRightConfig } from './kernels/FlipLeftRight' ; import { floorConfig } from './kernels/Floor' ; import { floorDivConfig } from './kernels/FloorDiv' ; import { fromPixelsConfig } from './kernels/FromPixels' ; import { fusedConv2DConfig } from './kernels/FusedConv2D' ; import { fusedDepthwiseConv2DConfig } from './kernels/FusedDepthwiseConv2D' ; import { gatherNdConfig } from './kernels/GatherNd' ; import { gatherV2Config } from './kernels/GatherV2' ; import { greaterConfig } from './kernels/Greater' ; import { greaterEqualConfig } from './kernels/GreaterEqual' ; import { identityConfig } from './kernels/Identity' ; import { ifftConfig } from './kernels/IFFT' ; import { imagConfig } from './kernels/Imag' ; import { isFiniteConfig } from './kernels/IsFinite' ; import { isInfConfig } from './kernels/IsInf' ; import { isNaNConfig } from './kernels/IsNaN' ; import { leakyReluConfig } from './kernels/LeakyRelu' ; import { lessConfig } from './kernels/Less' ; import { lessEqualConfig } from './kernels/LessEqual' ; import { linSpaceConfig } from './kernels/LinSpace' ; import { logConfig } from './kernels/Log' ; import { log1pConfig } from './kernels/Log1p' ; import { logicalAndConfig } from './kernels/LogicalAnd' ; import { logicalNotConfig } from './kernels/LogicalNot' ; import { logicalOrConfig } from './kernels/LogicalOr' ; import { LRNConfig } from './kernels/LRN' ; import { LRNGradConfig } from './kernels/LRNGrad' ; import { maxConfig } from './kernels/Max' ; import { maximumConfig } from './kernels/Maximum' ; import { maxPoolConfig } from './kernels/MaxPool' ; import { maxPool3DConfig } from './kernels/MaxPool3D' ; import { maxPool3DGradConfig } from './kernels/MaxPool3DGrad' ; import { maxPoolGradConfig } from './kernels/MaxPoolGrad' ; import { maxPoolWithArgmaxConfig } from './kernels/MaxPoolWithArgmax' ; import { meanConfig } from './kernels/Mean' ; import { minConfig } from './kernels/Min' ; import { minimumConfig } from './kernels/Minimum' ; import { mirrorPadConfig } from './kernels/MirrorPad' ; import { modConfig } from './kernels/Mod' ; import { multinomialConfig } from './kernels/Multinomial' ; import { multiplyConfig } from './kernels/Multiply' ; import { negConfig } from './kernels/Neg' ; import { nonMaxSuppressionV3Config } from './kernels/NonMaxSuppressionV3' ; import { nonMaxSuppressionV4Config } from './kernels/NonMaxSuppressionV4' ; import { nonMaxSuppressionV5Config } from './kernels/NonMaxSuppressionV5' ; import { notEqualConfig } from './kernels/NotEqual' ; import { oneHotConfig } from './kernels/OneHot' ; import { onesLikeConfig } from './kernels/OnesLike' ; import { packConfig } from './kernels/Pack' ; import { padV2Config } from './kernels/PadV2' ; import { powConfig } from './kernels/Pow' ; import { preluConfig } from './kernels/Prelu' ; import { prodConfig } from './kernels/Prod' ; import { rangeConfig } from './kernels/Range' ; import { realConfig } from './kernels/Real' ; import { realDivConfig } from './kernels/RealDiv' ; import { reciprocalConfig } from './kernels/Reciprocal' ; import { reluConfig } from './kernels/Relu' ; import { relu6Config } from './kernels/Relu6' ; import { reshapeConfig } from './kernels/Reshape' ; import { resizeBilinearConfig } from './kernels/ResizeBilinear' ; import { resizeBilinearGradConfig } from './kernels/ResizeBilinearGrad' ; import { resizeNearestNeighborConfig } from './kernels/ResizeNearestNeighbor' ; import { resizeNearestNeighborGradConfig } from './kernels/ResizeNearestNeighborGrad' ; import { reverseConfig } from './kernels/Reverse' ; import { rotateWithOffsetConfig } from './kernels/RotateWithOffset' ; import { roundConfig } from './kernels/Round' ; import { rsqrtConfig } from './kernels/Rsqrt' ; import { scatterNdConfig } from './kernels/ScatterNd' ; import { selectConfig } from './kernels/Select' ; import { seluConfig } from './kernels/Selu' ; import { sigmoidConfig } from './kernels/Sigmoid' ; import { signConfig } from './kernels/Sign' ; import { sinConfig } from './kernels/Sin' ; import { sinhConfig } from './kernels/Sinh' ; import { sliceConfig } from './kernels/Slice' ; import { softmaxConfig } from './kernels/Softmax' ; import { softplusConfig } from './kernels/Softplus' ; import { spaceToBatchNDConfig } from './kernels/SpaceToBatchND' ; import { sparseFillEmptyRowsConfig } from './kernels/SparseFillEmptyRows' ; import { sparseReshapeConfig } from './kernels/SparseReshape' ; import { sparseSegmentMeanConfig } from './kernels/SparseSegmentMean' ; import { sparseSegmentSumConfig } from './kernels/SparseSegmentSum' ; import { sparseToDenseConfig } from './kernels/SparseToDense' ; import { splitVConfig } from './kernels/SplitV' ; import { sqrtConfig } from './kernels/Sqrt' ; import { squareConfig } from './kernels/Square' ; import { squaredDifferenceConfig } from './kernels/SquaredDifference' ; import { stepConfig } from './kernels/Step' ; import { stridedSliceConfig } from './kernels/StridedSlice' ; import { stringNGramsConfig } from './kernels/StringNGrams' ; import { stringSplitConfig } from './kernels/StringSplit' ; import { stringToHashBucketFastConfig } from './kernels/StringToHashBucketFast' ; import { subConfig } from './kernels/Sub' ; import { sumConfig } from './kernels/Sum' ; import { tanConfig } from './kernels/Tan' ; import { tanhConfig } from './kernels/Tanh' ; import { tileConfig } from './kernels/Tile' ; import { topKConfig } from './kernels/TopK' ; import { transformConfig } from './kernels/Transform' ; import { transposeConfig } from './kernels/Transpose' ; import { uniqueConfig } from './kernels/Unique' ; import { unpackConfig } from './kernels/Unpack' ; import { unsortedSegmentSumConfig } from './kernels/UnsortedSegmentSum' ; import { zerosLikeConfig } from './kernels/ZerosLike' ; const kernelConfigs = [ _fusedMatMulConfig , absConfig , acosConfig , acoshConfig , addConfig , addNConfig , allConfig , anyConfig , argMaxConfig , argMinConfig , asinConfig , asinhConfig , atanConfig , atan2Config , atanhConfig , avgPoolConfig , avgPool3DConfig , avgPool3DGradConfig , avgPoolGradConfig , batchMatMulConfig , batchNormConfig , batchToSpaceNDConfig , bincountConfig , broadcastArgsConfig , castConfig , ceilConfig , clipByValueConfig , complexConfig , complexAbsConfig , concatConfig , conv2DConfig , conv2DBackpropFilterConfig , conv2DBackpropInputConfig , conv3DConfig , conv3DBackpropFilterV2Config , conv3DBackpropInputConfig , cosConfig , coshConfig , cropAndResizeConfig , cumsumConfig , denseBincountConfig , depthToSpaceConfig , depthwiseConv2dNativeConfig , depthwiseConv2dNativeBackpropFilterConfig , depthwiseConv2dNativeBackpropInputConfig , diagConfig , dilation2DConfig , einsumConfig , eluConfig , eluGradConfig , equalConfig , erfConfig , expConfig , expandDimsConfig , expm1Config , fftConfig , fillConfig , flipLeftRightConfig , floorConfig , floorDivConfig , fromPixelsConfig , fusedConv2DConfig , fusedDepthwiseConv2DConfig , gatherNdConfig , gatherV2Config , greaterConfig , greaterEqualConfig , identityConfig , ifftConfig , imagConfig , isFiniteConfig , isInfConfig , isNaNConfig , leakyReluConfig , lessConfig , lessEqualConfig , linSpaceConfig , logConfig , log1pConfig , logicalAndConfig , logicalNotConfig , logicalOrConfig , LRNConfig , LRNGradConfig , maxConfig , maximumConfig , maxPoolConfig , maxPool3DConfig , maxPool3DGradConfig , maxPoolGradConfig , maxPoolWithArgmaxConfig , meanConfig , minConfig , minimumConfig , mirrorPadConfig , modConfig , multinomialConfig , multiplyConfig , negConfig , nonMaxSuppressionV3Config , nonMaxSuppressionV4Config , nonMaxSuppressionV5Config , notEqualConfig , oneHotConfig , onesLikeConfig , packConfig , padV2Config , powConfig , preluConfig , prodConfig , rangeConfig , realConfig , realDivConfig , reciprocalConfig , reluConfig , relu6Config , reshapeConfig , resizeBilinearConfig , resizeBilinearGradConfig , resizeNearestNeighborConfig , resizeNearestNeighborGradConfig , reverseConfig , rotateWithOffsetConfig , roundConfig , rsqrtConfig , scatterNdConfig , selectConfig , seluConfig , sigmoidConfig , signConfig , sinConfig , sinhConfig , sliceConfig , softmaxConfig , softplusConfig , spaceToBatchNDConfig , sparseFillEmptyRowsConfig , sparseReshapeConfig , sparseSegmentMeanConfig , sparseSegmentSumConfig , sparseToDenseConfig , splitVConfig , sqrtConfig , squareConfig , squaredDifferenceConfig , stepConfig , stridedSliceConfig , stringNGramsConfig , stringSplitConfig , stringToHashBucketFastConfig , subConfig , sumConfig , tanConfig , tanhConfig , tileConfig , topKConfig , transformConfig , transposeConfig , uniqueConfig , unpackConfig , unsortedSegmentSumConfig , zerosLikeConfig ] ; for ( const kernelConfig of kernelConfigs ) { registerKernel ( kernelConfig ) ; }