| 142 | class RandomGammaOp : public OpKernel { |
| 143 | public: |
| 144 | explicit RandomGammaOp(OpKernelConstruction* context) : OpKernel(context) { |
| 145 | OP_REQUIRES_OK(context, generator_.Init(context)); |
| 146 | } |
| 147 | |
| 148 | void Compute(OpKernelContext* ctx) override { |
| 149 | const Tensor& shape_t = ctx->input(0); |