HomeSort by relevance Sort by last modified time
    Searched refs:im2col_buffer (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
quantized_conv_ops.cc 282 "Conv2d", "im2col_buffer",
290 T1* im2col_buffer = im2col_buffer_resource->data; local
311 im2col_buffer + (patch_index_within_chunk * filter_value_count);
391 meta::QuantizedGemm(context, transpose_a, transpose_b, im2col_buffer,
400 const uint8* im2col_data_as_uint8 = &(im2col_buffer->value);
440 transpose_a, transpose_b, transpose_c, m, n, k, im2col_buffer,
conv_ops_using_gemm.cc 315 "Conv2d", "im2col_buffer",
323 T1* im2col_buffer = im2col_buffer_resource->data; local
343 im2col_buffer + (patch_index_within_chunk * filter_value_count);
416 gemm_functor(context, m, n, k, im2col_buffer, lda, filter_data, ldb,
conv_ops_fused.cc 348 "Conv2d", "im2col_buffer",
395 T1* im2col_buffer = im2col_buffer_resource->data; local
546 im2col_buffer + (patch_index_within_chunk * filter_value_count);
592 gemm_functor(context, m, n, k, im2col_buffer, lda, filter_data, ldb,
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
multithreaded_conv.h 116 void operator()(const T* input_data, T* im2col_buffer, int input_batches,

Completed in 83 milliseconds