Method Compute

en_ops/e_matadd.cc:48–71

        OP_REQUIRES(context, lib != NULL, errors::Unknown("Unable to load sgx.so!"));
    }
    void Compute(OpKernelContext *context) override
    {
        const Tensor &input1 = context->input(0);
        auto input1_flat = input1.flat<float>();

        const Tensor &input2 = context->input(1);
        auto input2_flat = input2.flat<float>();

        const TensorShape &input_shape = input1.shape();

        Tensor *output = NULL;
        OP_REQUIRES_OK(context, context->allocate_output(0, input_shape, &output));
        auto output_flat = output->flat<float>();
        int N = input1_flat.size() / times_;
        int C = input2_flat.size();

        unsigned long int eid_ = (eid_high_ << 32) + eid_low_;
        typedef void (*function)(unsigned long int eid, float *input1, float *input2, int N, int C, float *output);
        dlerror();
        function matadd_kernel = (function)dlsym(lib, "matadd");
        const char *dlsym_error = dlerror();
        OP_REQUIRES(context, !dlsym_error, errors::Unknown("loading of matadd failed: ", dlsym_error));
        matadd_kernel(eid_, (float *)input1_flat.data(), (float *)input2_flat.data(), N, C, (float *)output_flat.data());
    };

private:
    void *lib;