Searched defs:im2col_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 140 float* output_data, const RuntimeShape& im2col_shape, in Conv()
|
| D | optimized_ops.h | 2184 float* output_data, const RuntimeShape& im2col_shape, in Conv()
2290 const RuntimeShape& im2col_shape, int8_t* im2col_data) { in HybridConv()
2367 uint8* output_data, const RuntimeShape& im2col_shape, in Conv()
6326 float* output_data, const RuntimeShape& im2col_shape, float* im2col_data) { in TransposeConv()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 137 float* output_data, const RuntimeShape& im2col_shape, in Conv()
210 uint8* output_data, const RuntimeShape& im2col_shape, in Conv()
3983 float* output_data, const RuntimeShape& im2col_shape, float* im2col_data) { in TransposeConv()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 2115 float* output_data, const RuntimeShape& im2col_shape, in Conv()
2208 const RuntimeShape& im2col_shape, int8_t* im2col_data) { in HybridConv()
2364 uint8* output_data, const RuntimeShape& im2col_shape, in Conv()
6516 float* output_data, const RuntimeShape& im2col_shape, float* im2col_data) { in TransposeConv()
|