Searched refs:col2im_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | transpose_conv.h | 30 int8_t* output_data, const RuntimeShape& col2im_shape, int32_t* col2im_data, in TransposeConvV2() argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 2952 float* output_data, const RuntimeShape& col2im_shape, float* col2im_data, in TransposeConvV2() argument
2957 col2im_shape, col2im_data, cpu_backend_context); in TransposeConvV2()
|
| D | optimized_ops.h | 5717 float* const output_data, const RuntimeShape& col2im_shape, in TransposeConvV2() argument
5942 uint8_t* output_data, const RuntimeShape& col2im_shape, in TransposeConvV2() argument
|