Searched refs:im2col_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 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()
565 const Dims<4>& im2col_dims) { in Conv() argument
572 im2col_dims); in Conv()
582 const Dims<4>& im2col_dims) { in Conv() argument
588 im2col_data, im2col_dims); in Conv()
598 const Dims<4>& im2col_dims) { in Conv() argument
601 output_dims, im2col_data, im2col_dims); 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()
235 const Dims<4>& im2col_dims) { in Conv() argument
242 im2col_dims); in Conv()
252 const Dims<4>& im2col_dims) { in Conv() argument
258 im2col_data, im2col_dims); in Conv()
268 const Dims<4>& im2col_dims) { 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 …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 2183 float* im2col_data, const Dims<4>& im2col_dims) { in Conv() argument
2198 output_data, DimsToShape(im2col_dims), im2col_data); in Conv()
2293 int8_t* im2col_data, const Dims<4>& im2col_dims) { in HybridConv() argument
2307 output_data, DimsToShape(im2col_dims), im2col_data); in HybridConv()
2319 const Dims<4>& im2col_dims) { in Conv() argument
2326 im2col_dims); in Conv()
2337 const Dims<4>& im2col_dims) { in Conv() argument
2343 im2col_data, im2col_dims); in Conv()
2354 const Dims<4>& im2col_dims) { in Conv() argument
2357 output_dims, im2col_data, im2col_dims); in Conv()
[all …]
|