Searched refs:gemm_output_buffer (Results 1 – 2 of 2) sorted by relevance
859 T* gemm_output_buffer) in Conv2DState()869 gemm_output_buffer(gemm_output_buffer) {} in Conv2DState()880 T* gemm_output_buffer; member916 cs.packed_tile_buffer, cs.gemm_output_buffer); in operator ()()922 memcpy(cs.buffer1 + i * gemm_out_buf_size, cs.gemm_output_buffer, in operator ()()1120 T* gemm_output_buffer = gemm_output_tensor.template flat<T>().data(); in operator ()() local1126 packed_tile_buffer, gemm_output_buffer); in operator ()()
806 gemm_output_buffer = template_buffer893 gemm_output_buffer = ir.Buffer(gemm_output_name, template_buffer.layout)894 current_input_buffer = gemm_output_buffer1013 GemmOut=gemm_output_buffer,