Method Compute

en_ops/encrypt.cc:87–127

        OP_REQUIRES(context, lib != NULL, errors::Unknown("Unable to load sgx.so!"));
    }
    void Compute(OpKernelContext *context) override
    {

        const Tensor &input = context->input(0);
        auto input_flat = input.flat<float>();
        // check shapes of input
        const TensorShape &input_shape = input.shape();
        TensorShape output_shape;
        if (input_shape.dims() == 2)
        {
            output_shape.AddDim(times_ * input_shape.dim_size(0));
            output_shape.AddDim(input_shape.dim_size(1));
        }
        else if (input_shape.dims() == 3)
        {
            output_shape.AddDim(times_ * input_shape.dim_size(0));
            output_shape.AddDim(input_shape.dim_size(1));
            output_shape.AddDim(input_shape.dim_size(2));
        }
        else
        {
            output_shape.AddDim(times_ * input_shape.dim_size(0));
            output_shape.AddDim(input_shape.dim_size(1));
            output_shape.AddDim(input_shape.dim_size(2));
            output_shape.AddDim(input_shape.dim_size(3));
        }
        // create output tensor
        Tensor *output = NULL;
        OP_REQUIRES_OK(context, context->allocate_output(0, output_shape, &output));
        auto output_flat = output->flat<float>();
        int N = input_flat.size();

        unsigned long int eid_ = (eid_high_ << 32) + eid_low_;
        typedef void (*function)(unsigned long int eid, float *x, int N, float *y);
        dlerror();
        function encrypt_kernel = (function)dlsym(lib, "encrypt1");
        const char *dlsym_error = dlerror();

        OP_REQUIRES(context, !dlsym_error, errors::Unknown("loading of encrypt1 failed: ", dlsym_error));
        encrypt_kernel(eid_, (float *)input_flat.data(), N, (float *)output_flat.data());
    };

private:
    void *lib;