DLPrimitives
|
This is the complete list of members for dlprim::gpu::GEMM, including all inherited members.
bias_M (defined in dlprim::gpu::GEMM) | dlprim::gpu::GEMM | static |
bias_N (defined in dlprim::gpu::GEMM) | dlprim::gpu::GEMM | static |
gemm(int M, int N, int K, cl::Buffer &a, cl_ulong offset_a, int lda, cl::Buffer &b, cl_ulong offset_b, int ldb, cl::Buffer &c, cl_ulong offset_c, int ldc, cl::Buffer *bias, cl_ulong bias_offset, float beta, int size_of_c, ExecutionContext const &e)=0 (defined in dlprim::gpu::GEMM) | dlprim::gpu::GEMM | pure virtual |
get_optimal_conv_gemm(Context &ctx, DataType dtype, GemmOpMode op_mode, bool trans_a, bool trans_b, int M, int N, int K, int kernel[2], int dilate[2], int padding[2], int stride[2], int groups, int src_channels, int src_rows, int src_cols, int tgt_rows, int tgt_cols, int bias=0, StandardActivations act=StandardActivations::identity, int im2col_chan=0) (defined in dlprim::gpu::GEMM) | dlprim::gpu::GEMM | static |
get_optimal_gemm(Context &ctx, DataType dtype, bool trans_a, bool trans_b, int M, int N, int K, int bias=0, StandardActivations act=StandardActivations::identity, int im2col_chan=0) (defined in dlprim::gpu::GEMM) | dlprim::gpu::GEMM | static |
no_bias (defined in dlprim::gpu::GEMM) | dlprim::gpu::GEMM | static |
~GEMM() (defined in dlprim::gpu::GEMM) | dlprim::gpu::GEMM | inlinevirtual |