Searched defs:im2col_data (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | conv.h | 32 float* im2col_data) { in Conv() 115 uint8_t* im2col_data, void* cpu_backend_context) { 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() 122 uint8_t* im2col_data, int32_t* scratch_buffer) { in TransposeConv()
|
| D | legacy_reference_ops.h | 215 float* im2col_data, const Dims<4>& im2col_dims) { in Conv() 239 float* output_data, const Dims<4>& output_dims, float* im2col_data, in Conv() 256 const Dims<4>& output_dims, float* im2col_data, in Conv() 272 const Dims<4>& output_dims, float* im2col_data, in Conv() 288 uint8* im2col_data, const Dims<4>& im2col_dims, in Conv() 321 const Dims<4>& output_dims, uint8* im2col_data, in Conv() 341 const Dims<4>& output_dims, uint8* im2col_data, in Conv() 369 uint8* output_data, const Dims<4>& output_dims, uint8* im2col_data, in Conv() 382 const Dims<4>& output_dims, float* im2col_data, in TransposeConv() 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/ |
| D | im2col_utils.h | 121 const RuntimeShape& output_shape, T* im2col_data, in DilatedIm2col() 207 const RuntimeShape& output_shape, T* im2col_data) { in DilatedIm2col() 377 T* im2col_data) { in Im2col3D() 422 const RuntimeShape& im2col_shape, T* im2col_data) { in DilatedIm2col3D()
|
| D | multithreaded_conv.h | 141 float* im2col_data) { in Conv()
|
| D | legacy_optimized_ops.h | 2379 T* im2col_data) { in DilatedIm2col() 2428 float* im2col_data) { in Conv() 2533 float* im2col_data, const Dims<4>& im2col_dims) { in Conv() 2559 int8_t* im2col_data, const Dims<4>& im2col_dims, in HybridConv() 2584 float* output_data, const Dims<4>& output_dims, float* im2col_data, in Conv() 2601 const Dims<4>& output_dims, float* im2col_data, in Conv() 2617 const Dims<4>& output_dims, float* im2col_data, in Conv() 2629 uint8* im2col_data, gemmlowp::GemmContext* gemmlowp_context) { in Conv() 2739 uint8* im2col_data, const Dims<4>& im2col_dims, in Conv() 2772 const Dims<4>& output_dims, uint8* im2col_data, in Conv() [all …]
|
| D | optimized_ops.h | 912 float* im2col_data, CpuBackendContext* cpu_backend_context) { in Conv() 1011 int8_t* im2col_data, CpuBackendContext* context) { in HybridConv() 1100 const RuntimeShape& im2col_shape, int8_t* im2col_data, in HybridConvPerChannel() 1210 uint8* im2col_data, CpuBackendContext* cpu_backend_context) { in Conv() 4824 const RuntimeShape& output_shape, T* im2col_data) { in TransposeIm2col() 8052 float* im2col_data, in Conv3D()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | conv.h | 39 InputScalar* im2col_data, CpuBackendContext* cpu_backend_context) { in ConvPerChannel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | conv_per_channel_quantized_16x8_test.cc | 285 float im2col_data; in TryTestOneConvFilter() local
|