Method Compile

tensorflow/compiler/tf2xla/kernels/relu_op.cc:42–44

Computes the max of the scalar input x and 0.

  explicit ReluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
  // Computes the max of the scalar input x and 0.
  void Compile(XlaOpKernelContext* ctx) override {
    ctx->SetOutput(0, xla::Relu(ctx->Input(0)));
  }
};
REGISTER_XLA_OP(Name("Relu"), ReluOp);