Searched defs:im2col_shape (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | conv.h | 31 float* output_data, const RuntimeShape& im2col_shape, in Conv()
114 uint8_t* output_data, const RuntimeShape& im2col_shape, in Conv()
209 const RuntimeShape& im2col_shape, int8_t* im2col_data, in HybridConvPerChannel()
|
| D | transpose_conv.h | 32 float* output_data, const RuntimeShape& im2col_shape, float* im2col_data) { in TransposeConv()
121 uint8_t* output_data, const RuntimeShape& im2col_shape, in TransposeConv()
|
| D | legacy_reference_ops.h | 403 float* output_data, const RuntimeShape& im2col_shape, float* im2col_data) { in TransposeConv()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | transpose_conv.h | 32 int8_t* output_data, const RuntimeShape& im2col_shape, int8_t* im2col_data, in TransposeConv()
132 int16_t* output_data, const RuntimeShape& im2col_shape, int8_t* im2col_data, in TransposeConv()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | conv.h | 38 DstScalar* output_data, const RuntimeShape& im2col_shape, in ConvPerChannel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 140 float* output_data, const RuntimeShape& im2col_shape, in Conv()
|
| D | im2col_utils.h | 376 const T* input_data, const RuntimeShape& im2col_shape, in Im2col3D()
422 const RuntimeShape& im2col_shape, T* im2col_data) { in DilatedIm2col3D()
|
| D | optimized_ops.h | 911 float* output_data, const RuntimeShape& im2col_shape, in Conv()
1010 float* output_data, const RuntimeShape& im2col_shape, in HybridConv()
1100 const RuntimeShape& im2col_shape, int8_t* im2col_data, in HybridConvPerChannel()
1209 uint8* output_data, const RuntimeShape& im2col_shape, in Conv()
8051 float* output_data, const RuntimeShape& im2col_shape, in Conv3D()
|
| D | legacy_optimized_ops.h | 2427 float* output_data, const RuntimeShape& im2col_shape, in Conv()
2628 uint8* output_data, const RuntimeShape& im2col_shape, in Conv()
2912 float* output_data, const RuntimeShape& im2col_shape, float* im2col_data) { in TransposeConv()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | conv_per_channel_quantized_16x8_test.cc | 284 RuntimeShape im2col_shape; in TryTestOneConvFilter() local
|