Method Compute

tensorflow/core/kernels/sparse_reshape_op.cc:50–53

    explicit SparseReshapeOpGPU(OpKernelConstruction *context) : OpKernel(context) {}

  void Compute(OpKernelContext* context) override {
    ReshapeGPU(context, context->input(0), context->input(1), context->input(2),
            0 /* output indices index */, 1 /* output shape index */);
  }
};

REGISTER_KERNEL_BUILDER(Name("SparseReshape")