D | GEMMFixture.h | 235 DataType data_type, GPUTarget gpu_arch) in setup() argument 244 …ape, bias_shape, data_type, alpha, beta, broadcast_bias, fp16_mixed_precision, act_info, gpu_arch); in setup() 264 … bool fp16_mixed_precision, const ActivationLayerInfo &act_info, GPUTarget gpu_arch) in compute_target() argument 281 …gemm.configure(gpu_arch, lhs.info(), rhs.info(), bias.info(), dst.info(), alpha, beta, false, resh… in compute_target() 359 const ActivationLayerInfo &act_info, DataType data_type, GPUTarget gpu_arch) in setup() argument 371 …ape, rhs_shape, bias_shape, data_type, alpha, beta, m_h, fp16_mixed_precision, act_info, gpu_arch); in setup() 387 … bool fp16_mixed_precision, const ActivationLayerInfo &act_info, GPUTarget gpu_arch) in compute_target() argument 404 …gemm.configure(gpu_arch, lhs.info(), rhs.info(), bias.info(), dst.info(), alpha, beta, false, resh… in compute_target() 481 const ActivationLayerInfo &act_info, DataType data_type, GPUTarget gpu_arch) in setup() argument 504 …_info, rhs_info, data_type, alpha, beta, broadcast_bias, fp16_mixed_precision, act_info, gpu_arch); in setup() [all …]
|