Home
last modified time | relevance | path

Searched refs:gemm_output_buffer (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddeep_conv2d.cc845 T* gemm_output_buffer) in Conv2DState()
855 gemm_output_buffer(gemm_output_buffer) {} in Conv2DState()
866 T* gemm_output_buffer; member
902 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 ()() local
1108 packed_tile_buffer, gemm_output_buffer); in operator ()()