Searched defs:linear_op (Results 1 – 5 of 5) sorted by relevance
83 xnn_operator_t linear_op{}; in create() local
59 def linear_op(act, weight, bias=None): function
177 auto linear_op = cudnn_frontend::OperationBuilder(CUDNN_BACKEND_OPERATION_MATMUL_DESCRIPTOR) in apply_impl_helper() local
70 #define linear_op(DATA_TYPE, VEC_SIZE, x, A_VAL, B_VAL) (MLA((DATA_TYPE)B_VAL, (DATA_TYPE)A_VAL, x)) macro
1415 def __init__(self, linear_op): argument