Method Compile

tensorflow/compiler/tf2xla/kernels/relu_op.cc:52–54

Clamp the scalar input between 0 and 6.

  explicit Relu6Op(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
  // Clamp the scalar input between 0 and 6.
  void Compile(XlaOpKernelContext* ctx) override {
    ctx->SetOutput(0, xla::Relu6(ctx->Input(0)));
  }
};
REGISTER_XLA_OP(Name("Relu6"), Relu6Op);