license import { backend_util , KernelConfig , Selu } from '@tensorflow/tfjs-core' ; import { unaryKernelFunc } from '../kernel_utils/kernel_funcs_utils' ; const SELU = `
  // Stable and Attracting Fixed Point (0, 1) for Normalized Weights.
  // see: https://arxiv.org/abs/1706.02515
  float scaleAlpha = ${ backend_util . SELU_SCALEALPHA };
  float scale = ${ backend_util . SELU_SCALE };
  return (x >= 0.0) ? scale * x : scaleAlpha * (exp(x) - 1.0);
` ; export const selu = unaryKernelFunc ( { opSnippet : SELU } ) ; export const seluConfig = { kernelName : Selu , backendName : 'webgl' , kernelFunc : selu , } ;