Searched refs:im2col_buffer (Results 1 – 4 of 4) sorted by relevance
| /external/XNNPACK/bench/ |
| D | f32-im2col-gemm.cc | 82 …std::vector<float> im2col_buffer(output_size * group_input_channels * kernel_size * group_output_c… in Im2ColGEMMBenchmark() local
107 a.data(), im2col_buffer.data()); in Im2ColGEMMBenchmark()
108 inputData = im2col_buffer.data(); in Im2ColGEMMBenchmark()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_conv_ops.cc | 290 T1* im2col_buffer = im2col_buffer_resource->data; in operator ()() local
311 im2col_buffer + (patch_index_within_chunk * filter_value_count); in operator ()()
391 meta::QuantizedGemm(context, transpose_a, transpose_b, im2col_buffer, in operator ()()
400 const uint8* im2col_data_as_uint8 = &(im2col_buffer->value); in operator ()()
440 transpose_a, transpose_b, transpose_c, m, n, k, im2col_buffer, in operator ()()
|
| D | conv_ops_using_gemm.cc | 325 T1* im2col_buffer = im2col_buffer_resource->data; in operator ()() local
345 im2col_buffer + (patch_index_within_chunk * filter_value_count); in operator ()()
418 gemm_functor(context, m, n, k, im2col_buffer, lda, filter_data, ldb, in operator ()()
|
| D | conv_ops_fused_image_transform.cc | 397 T1* im2col_buffer = im2col_buffer_resource->data; in operator ()() local
548 im2col_buffer + (patch_index_within_chunk * filter_value_count); in operator ()()
594 gemm_functor(context, m, n, k, im2col_buffer, lda, filter_data, ldb, in operator ()()
|