Searched defs:kwidth (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | im2col_utils.h | 29 int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn()
214 void Im2col(const ConvParams& params, int kheight, int kwidth, uint8 zero_byte, in Im2col()
249 void Im2col(const ConvParams& params, int kheight, int kwidth, in Im2col()
289 int kdepth, int kheight, int kwidth, // Kernel params. in ExtractPatchIntoBufferColumn3D()
374 void Im2col3D(const Conv3DParams& params, int kdepth, int kheight, int kwidth, in Im2col3D()
|
| H A D | legacy_optimized_ops.h | 2363 const Dims<4>& input_dims, int w, int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn()
2398 int kwidth, uint8 zero_byte, T* output_data, in Im2col()
2417 int pad_width, int pad_height, int kheight, int kwidth, in Im2col()
2837 int pad_width, int pad_height, int kheight, int kwidth, in Im2col()
3768 int pad_height, int kwidth, int kheight, in AveragePool()
3789 int pad_height, int kwidth, int kheight, float* output_data, in AveragePool()
3867 int pad_height, int kwidth, int kheight, in MaxPool()
3887 int kwidth, int kheight, float* output_data, in MaxPool()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | create_im2col_arrays.cc | 42 const int kwidth = weights_shape.dims(2); in ProcessConvOperator() local
|
| H A D | propagate_fixed_sizes.cc | 35 void ComputeConvSizes(const Shape& input_shape, int output_depth, int kwidth, in ComputeConvSizes()
147 const int kwidth = weights_shape.dims(2); in ProcessConvOperator() local
209 const int kwidth = weights_shape.dims(2); in ProcessTransposeConvOperator() local
286 const int kwidth = weights_shape.dims(2); in ProcessDepthwiseConvOperator() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | legacy_reference_ops.h | 1494 int pad_height, int kwidth, int kheight, in AveragePool()
1569 int pad_height, int kwidth, int kheight, float* output_data, in AveragePool()
1647 int pad_height, int kwidth, int kheight, in MaxPool()
1667 int kwidth, int kheight, float* output_data, in MaxPool()
|