Searched refs:gemm_out_buf_size (Results 1 – 1 of 1) sorted by relevance
895 const int64 gemm_out_buf_size = num_tiles * num_filters; in operator ()() local896 const int64 gemm_out_buf_bytes = gemm_out_buf_size * sizeof(T); in operator ()()899 GemmState<T> gemm(num_filters, num_tiles, in_depth, gemm_out_buf_size, in operator ()()908 memcpy(cs.buffer1 + i * gemm_out_buf_size, cs.gemm_output_buffer, in operator ()()