| 67 | public: |
| 68 | typedef typename Distribution::ResultElementType T; |
| 69 | explicit PhiloxRandomOp(OpKernelConstruction* ctx) : OpKernel(ctx) { |
| 70 | OP_REQUIRES_OK(ctx, generator_.Init(ctx)); |
| 71 | } |
| 72 | |
| 73 | void Compute(OpKernelContext* ctx) override { |
| 74 | const Tensor& shape = ctx->input(0); |