Searched defs: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()
541 int8_t* im2col_data, const Dims<4>& im2col_dims) { in HybridConv()
565 const Dims<4>& im2col_dims) { in Conv()
582 const Dims<4>& im2col_dims) { in Conv()
598 const Dims<4>& im2col_dims) { in Conv()
613 uint8* im2col_data, const Dims<4>& im2col_dims, in Conv()
647 const Dims<4>& im2col_dims, in Conv()
667 const Dims<4>& im2col_dims, in Conv()
695 const Dims<4>& im2col_dims, gemmlowp::GemmContext* gemm_context) { in Conv()
787 const Dims<4>& im2col_dims) { in TransposeConv()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 210 float* im2col_data, const Dims<4>& im2col_dims) { in Conv()
235 const Dims<4>& im2col_dims) { in Conv()
252 const Dims<4>& im2col_dims) { in Conv()
268 const Dims<4>& im2col_dims) { in Conv()
283 uint8* im2col_data, const Dims<4>& im2col_dims, in Conv()
317 const Dims<4>& im2col_dims, in Conv()
337 const Dims<4>& im2col_dims, in Conv()
365 const Dims<4>& im2col_dims, gemmlowp::GemmContext* gemm_context) { in Conv()
378 const Dims<4>& im2col_dims) { in TransposeConv()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 2183 float* im2col_data, const Dims<4>& im2col_dims) { in Conv()
2293 int8_t* im2col_data, const Dims<4>& im2col_dims) { in HybridConv()
2319 const Dims<4>& im2col_dims) { in Conv()
2337 const Dims<4>& im2col_dims) { in Conv()
2354 const Dims<4>& im2col_dims) { in Conv()
2462 uint8* im2col_data, const Dims<4>& im2col_dims, in Conv()
2498 const Dims<4>& im2col_dims, in Conv()
2519 const Dims<4>& im2col_dims, in Conv()
2548 const Dims<4>& im2col_dims, gemmlowp::GemmContext* gemm_context) { in Conv()
6542 const Dims<4>& im2col_dims) { in TransposeConv()
|