Method Compute

en_ops/e_matmul2d.cc:105–138

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

        const Tensor &grad = context->input(0);
        const Tensor &input = context->input(1);
        const Tensor &weight = context->input(2);

        auto grad_flat = grad.flat<float>();
        auto input_flat = input.flat<float>();
        auto weight_flat = weight.flat<float>();

        const TensorShape &input_shape = input.shape();
        const TensorShape &weight_shape = weight.shape();

        Tensor *output1 = NULL;
        OP_REQUIRES_OK(context, context->allocate_output(0, input_shape, &output1));
        auto output1_flat = output1->flat<float>();

        Tensor *output2 = NULL;
        OP_REQUIRES_OK(context, context->allocate_output(1, weight_shape, &output2));
        auto output2_flat = output2->flat<float>();

        int N = input_shape.dim_size(0) / times_;
        int M = input_shape.dim_size(1);
        int C = weight_shape.dim_size(1);

        unsigned long int eid_ = (eid_high_ << 32) + eid_low_;
        typedef void (*function)(unsigned long int eid, float *input, float *grad, float *weight, int N, int M, int C, float *output1, float *output2);
        dlerror();
        function matmul2d_grad_kernel = (function)dlsym(lib, "matmul2d_grad");
        const char *dlsym_error = dlerror();
        OP_REQUIRES(context, !dlsym_error, errors::Unknown("loading of matmul2d_grad failed: ", dlsym_error));
        matmul2d_grad_kernel(eid_, (float *)input_flat.data(), (float *)grad_flat.data(), (float *)weight_flat.data(), N, M, C, (float *)output1_flat.data(), (float *)output2_flat.data());
    };

private:
    void *lib;