Searched refs:gemm_output_buffer (Results 1 – 1 of 1) sorted by relevance
845 T* gemm_output_buffer) in Conv2DState()855 gemm_output_buffer(gemm_output_buffer) {} in Conv2DState()866 T* gemm_output_buffer; member902 cs.packed_tile_buffer, cs.gemm_output_buffer); in operator ()()908 memcpy(cs.buffer1 + i * gemm_out_buf_size, cs.gemm_output_buffer, in operator ()()1102 T* gemm_output_buffer = gemm_output_tensor.template flat<T>().data(); in operator ()() local1108 packed_tile_buffer, gemm_output_buffer); in operator ()()