Searched refs: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() argument
|
| D | optimized_ops.h | 2099 const RuntimeShape im2col_shape( in DilatedIm2col()
2121 Offset(im2col_shape, 0, 0, row_offset, col_offset); in DilatedIm2col()
2136 Offset(im2col_shape, 0, 0, row_offset, col_offset); in DilatedIm2col()
2184 float* output_data, const RuntimeShape& im2col_shape, in Conv() argument
2197 (void)im2col_shape; in Conv()
2214 gemm_input_shape = &im2col_shape; in Conv()
2218 input_data, im2col_shape, im2col_data); in Conv()
2220 gemm_input_shape = &im2col_shape; in Conv()
2290 const RuntimeShape& im2col_shape, int8_t* im2col_data) { in HybridConv() argument
2314 input_data, im2col_shape, im2col_data); in HybridConv()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1976 const RuntimeShape im2col_shape( in DilatedIm2col()
1998 Offset(im2col_shape, 0, 0, row_offset, col_offset); in DilatedIm2col()
2013 Offset(im2col_shape, 0, 0, row_offset, col_offset); in DilatedIm2col()
2115 float* output_data, const RuntimeShape& im2col_shape, in Conv() argument
2128 (void)im2col_shape; in Conv()
2145 gemm_input_shape = &im2col_shape; in Conv()
2149 input_data, im2col_shape, im2col_data); in Conv()
2151 gemm_input_shape = &im2col_shape; in Conv()
2208 const RuntimeShape& im2col_shape, int8_t* im2col_data) { in HybridConv() argument
2232 input_data, im2col_shape, im2col_data); in HybridConv()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 137 float* output_data, const RuntimeShape& im2col_shape, in Conv() argument
152 (void)im2col_shape; // only used in optimized code. in Conv()
210 uint8* output_data, const RuntimeShape& im2col_shape, in Conv() argument
213 (void)im2col_shape; // only used in optimized code. in Conv()
3983 float* output_data, const RuntimeShape& im2col_shape, float* im2col_data) { in TransposeConv() argument
3992 (void)im2col_shape; // only used in optimized code. in TransposeConv()
|