| 34 | explicit SparseReshapeOp(OpKernelConstruction* context) : OpKernel(context) {} |
| 35 | |
| 36 | void Compute(OpKernelContext* context) override { |
| 37 | Reshape(context, context->input(0), context->input(1), context->input(2), |
| 38 | 0 /* output indices index */, 1 /* output shape index */); |
| 39 | } |
| 40 | }; |
| 41 | |
| 42 | REGISTER_KERNEL_BUILDER(Name("SparseReshape").Device(DEVICE_CPU), |