/**
|
* @license
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
* =============================================================================
|
*/
|
// We explicitly import the modular kernels so they get registered in the
|
// global registry when we compile the library. A modular build would replace
|
// the contents of this file and import only the kernels that are needed.
|
import {KernelConfig, registerKernel} from '@tensorflow/tfjs';
|
|
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 {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 {conv3DBackpropInputV2Config} from './kernels/Conv3DBackpropInputV2';
|
import {cosConfig} from './kernels/Cos';
|
import {coshConfig} from './kernels/Cosh';
|
import {cropAndResizeConfig} from './kernels/CropAndResize';
|
import {cumprodConfig} from './kernels/Cumprod';
|
import {cumsumConfig} from './kernels/Cumsum';
|
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 {dilation2dBackpropFilterConfig} from './kernels/Dilation2DBackpropFilter';
|
import {dilation2dBackpropInputConfig} from './kernels/Dilation2DBackpropInput';
|
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 {fusedBatchNormConfig} from './kernels/FusedBatchNorm';
|
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 {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 {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 {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 {staticRegexReplaceConfig} from './kernels/StaticRegexReplace';
|
import {stepConfig} from './kernels/Step';
|
import {stridedSliceConfig} from './kernels/StridedSlice';
|
import {subConfig} from './kernels/Sub';
|
import {sumConfig} from './kernels/Sum';
|
import {tanConfig} from './kernels/Tan';
|
import {tanhConfig} from './kernels/Tanh';
|
import {tensorScatterUpdateConfig} from './kernels/TensorScatterUpdate';
|
import {tileConfig} from './kernels/Tile';
|
import {topKConfig} from './kernels/TopK';
|
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';
|
|
// List all kernel configs here
|
const kernelConfigs: KernelConfig[] = [
|
FFTConfig,
|
IFFTConfig,
|
LRNConfig,
|
LRNGradConfig,
|
_fusedMatMulConfig,
|
absConfig,
|
acosConfig,
|
acoshConfig,
|
addConfig,
|
addNConfig,
|
allConfig,
|
anyConfig,
|
argMaxConfig,
|
argMinConfig,
|
asinConfig,
|
asinhConfig,
|
atan2Config,
|
atanConfig,
|
atanhConfig,
|
avgPool3DConfig,
|
avgPool3DGradConfig,
|
avgPoolConfig,
|
avgPoolGradConfig,
|
batchMatMulConfig,
|
batchToSpaceNDConfig,
|
broadcastArgsConfig,
|
castConfig,
|
ceilConfig,
|
clipByValueConfig,
|
complexAbsConfig,
|
complexConfig,
|
concatConfig,
|
conv2DBackpropFilterConfig,
|
conv2DBackpropInputConfig,
|
conv2DConfig,
|
conv3DBackpropFilterV2Config,
|
conv3DBackpropInputV2Config,
|
conv3DConfig,
|
cosConfig,
|
coshConfig,
|
cropAndResizeConfig,
|
cumprodConfig,
|
cumsumConfig,
|
bincountConfig,
|
depthToSpaceConfig,
|
depthwiseConv2dNativeBackpropFilterConfig,
|
depthwiseConv2dNativeBackpropInputConfig,
|
depthwiseConv2dNativeConfig,
|
diagConfig,
|
dilation2dBackpropFilterConfig,
|
dilation2dBackpropInputConfig,
|
dilation2dConfig,
|
eluConfig,
|
eluGradConfig,
|
einsumConfig,
|
equalConfig,
|
erfConfig,
|
expConfig,
|
expandDimsConfig,
|
expm1Config,
|
fillConfig,
|
flipLeftRightConfig,
|
floorConfig,
|
floorDivConfig,
|
fusedBatchNormConfig,
|
fusedConv2DConfig,
|
fusedDepthwiseConv2DConfig,
|
gatherNdConfig,
|
gatherV2Config,
|
greaterConfig,
|
greaterEqualConfig,
|
identityConfig,
|
imagConfig,
|
isFiniteConfig,
|
isInfConfig,
|
isNanConfig,
|
leakyReluConfig,
|
lessConfig,
|
lessEqualConfig,
|
linSpaceConfig,
|
log1pConfig,
|
logConfig,
|
logicalAndConfig,
|
logicalNotConfig,
|
logicalOrConfig,
|
maxConfig,
|
maxPool3DConfig,
|
maxPool3DGradConfig,
|
maxPoolConfig,
|
maxPoolGradConfig,
|
maximumConfig,
|
meanConfig,
|
minConfig,
|
minimumConfig,
|
mirrorPadConfig,
|
modConfig,
|
multinomialConfig,
|
multiplyConfig,
|
negConfig,
|
nonMaxSuppressionV3Config,
|
nonMaxSuppressionV4Config,
|
nonMaxSuppressionV5Config,
|
notEqualConfig,
|
oneHotConfig,
|
onesLikeConfig,
|
packConfig,
|
padV2Config,
|
powConfig,
|
preluConfig,
|
prodConfig,
|
rangeConfig,
|
realConfig,
|
realDivConfig,
|
reciprocalConfig,
|
relu6Config,
|
reluConfig,
|
reshapeConfig,
|
resizeBilinearConfig,
|
resizeBilinearGradConfig,
|
resizeNearestNeighborConfig,
|
resizeNearestNeighborGradConfig,
|
reverseConfig,
|
roundConfig,
|
rsqrtConfig,
|
scatterNdConfig,
|
selectConfig,
|
seluConfig,
|
sigmoidConfig,
|
signConfig,
|
sinConfig,
|
sinhConfig,
|
sliceConfig,
|
softmaxConfig,
|
softplusConfig,
|
spaceToBatchNDConfig,
|
sparseToDenseConfig,
|
splitVConfig,
|
sqrtConfig,
|
squareConfig,
|
squaredDifferenceConfig,
|
staticRegexReplaceConfig,
|
stepConfig,
|
stridedSliceConfig,
|
subConfig,
|
sumConfig,
|
tanConfig,
|
tanhConfig,
|
tensorScatterUpdateConfig,
|
tileConfig,
|
topKConfig,
|
transposeConfig,
|
uniqueConfig,
|
unpackConfig,
|
unsortedSegmentSumConfig,
|
zerosLikeConfig
|
];
|
|
for (const kernelConfig of kernelConfigs) {
|
registerKernel(kernelConfig);
|
}
|
