Searched refs:_gemm_function (Results 1 – 2 of 2) sorted by relevance
136 : _gemm_function(std::make_unique<CpuGemm>()), in CpuWinogradConv2d()259 …_gemm_function->configure(&_winograd_transformed_input, &_winograd_transformed_weights, nullptr, &… in configure()271 auto asm_mem_req = _gemm_function->workspace(); in configure()356 _gemm_function->run(gemm_pack); in run()418 _gemm_function->prepare(gemm_pack); in prepare()
111 std::unique_ptr<CpuGemm> _gemm_function; variable