| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 465 T* im2col_data) { in DilatedIm2col() argument
478 im2col_data); in DilatedIm2col()
516 float* im2col_data, const Dims<4>& im2col_dims) { in Conv() argument
531 output_data, DimsToShape(im2col_dims), im2col_data); in Conv()
541 int8_t* im2col_data, const Dims<4>& im2col_dims) { in HybridConv() argument
555 output_data, DimsToShape(im2col_dims), im2col_data); in HybridConv()
564 float* output_data, const Dims<4>& output_dims, float* im2col_data, in Conv() argument
571 output_activation_max, output_data, output_dims, im2col_data, in Conv()
581 const Dims<4>& output_dims, float* im2col_data, in Conv() argument
588 im2col_data, im2col_dims); in Conv()
[all …]
|
| D | multithreaded_conv.h | 141 float* im2col_data) { in Conv() argument
163 conv_functor(device, input_data, im2col_data, batches, input_height, in Conv()
|
| D | optimized_ops.h | 2066 const RuntimeShape& output_shape, T* im2col_data) { in DilatedIm2col() argument
2082 TFLITE_DCHECK(im2col_data); in DilatedIm2col()
2120 T* dst = im2col_data + in DilatedIm2col()
2135 T* dst = im2col_data + in DilatedIm2col()
2185 float* im2col_data) { in Conv() argument
2196 (void)im2col_data; in Conv()
2212 filter_shape, output_shape, im2col_data); in Conv()
2213 gemm_input_data = im2col_data; in Conv()
2216 TFLITE_DCHECK(im2col_data); in Conv()
2218 input_data, im2col_shape, im2col_data); in Conv()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1943 const RuntimeShape& output_shape, T* im2col_data) { in DilatedIm2col() argument
1959 TFLITE_DCHECK(im2col_data); in DilatedIm2col()
1997 T* dst = im2col_data + in DilatedIm2col()
2012 T* dst = im2col_data + in DilatedIm2col()
2030 T* im2col_data) { in DilatedIm2col() argument
2043 im2col_data); in DilatedIm2col()
2116 float* im2col_data) { in Conv() argument
2127 (void)im2col_data; in Conv()
2143 filter_shape, output_shape, im2col_data); in Conv()
2144 gemm_input_data = im2col_data; in Conv()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 210 float* im2col_data, const Dims<4>& im2col_dims) { in Conv() argument
225 output_data, DimsToShape(im2col_dims), im2col_data); in Conv()
234 float* output_data, const Dims<4>& output_dims, float* im2col_data, in Conv() argument
241 output_activation_max, output_data, output_dims, im2col_data, in Conv()
251 const Dims<4>& output_dims, float* im2col_data, in Conv() argument
258 im2col_data, im2col_dims); in Conv()
267 const Dims<4>& output_dims, float* im2col_data, in Conv() argument
271 output_dims, im2col_data, im2col_dims); in Conv()
283 uint8* im2col_data, const Dims<4>& im2col_dims, in Conv() argument
305 output_data, DimsToShape(im2col_dims), im2col_data, gemm_context); in Conv()
[all …]
|
| D | reference_ops.h | 138 float* im2col_data) { in Conv() argument
151 (void)im2col_data; // only used in optimized code. in Conv()
211 uint8* im2col_data, gemmlowp::GemmContext* gemm_context) { in Conv() argument
212 (void)im2col_data; // only used in optimized code. in Conv()
3983 float* output_data, const RuntimeShape& im2col_shape, float* im2col_data) { in TransposeConv() argument
3991 (void)im2col_data; // only used in optimized code. in TransposeConv()
|