HomeSort by relevance Sort by last modified time
    Searched refs:SeluOp (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
elu_op.cc 64 class SeluOp : public XlaOpKernel {
66 explicit SeluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
105 REGISTER_XLA_OP(Name("Selu"), SeluOp);
  /external/tensorflow/tensorflow/core/kernels/
relu_op.cc 62 SeluOp<CPUDevice, type>); \
152 SeluOp<GPUDevice, type>); \
185 SeluOp<SYCLDevice, type>); \
relu_op.h 177 class SeluOp : public UnaryElementWiseOp<T, SeluOp<Device, T>> {
179 using UnaryElementWiseOp<T, SeluOp<Device, T>>::UnaryElementWiseOp;
198 // a (outputs): outputs of the SeluOp()

Completed in 478 milliseconds