Searched refs:gemm_output_buffer (Results 1 – 1 of 1) sorted by relevance
858 T* gemm_output_buffer) in Conv2DState()868 gemm_output_buffer(gemm_output_buffer) {} in Conv2DState()879 T* gemm_output_buffer; member915 cs.packed_tile_buffer, cs.gemm_output_buffer); in operator ()()921 memcpy(cs.buffer1 + i * gemm_out_buf_size, cs.gemm_output_buffer, in operator ()()1119 T* gemm_output_buffer = gemm_output_tensor.template flat<T>().data(); in operator ()() local1125 packed_tile_buffer, gemm_output_buffer); in operator ()()