tOp : public OpKernel {  public:   explicit ZeroOutOp(OpKernelConstruction* context) : OpKernel(context) {}    void Compute(OpKernelContext* context) override {     // Grab the input tensor     const Tensor& input_tensor = context->input(0);     