Searched defs:kheight (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | create_im2col_arrays.cc | 41 const int kheight = weights_shape.dims(1); in ProcessConvOperator() local
|
D | propagate_fixed_sizes.cc | 36 int kheight, int stride_width, int stride_height, in ComputeConvSizes() 146 const int kheight = weights_shape.dims(1); in ProcessConvOperator() local 210 const int kheight = weights_shape.dims(1); in ProcessTransposeConvOperator() local 287 const int kheight = weights_shape.dims(1); in ProcessDepthwiseConvOperator() local
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | im2col_utils.h | 28 int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn() 213 void Im2col(const ConvParams& params, int kheight, int kwidth, uint8 zero_byte, in Im2col() 248 void Im2col(const ConvParams& params, int kheight, int kwidth, in Im2col()
|
D | legacy_optimized_ops.h | 2362 const Dims<4>& input_dims, int w, int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn() 2396 int stride_height, int pad_width, int pad_height, int kheight, in Im2col() 2416 int pad_width, int pad_height, int kheight, int kwidth, in Im2col() 2836 int pad_width, int pad_height, int kheight, int kwidth, in Im2col() 3767 int pad_height, int kwidth, int kheight, in AveragePool() 3788 int pad_height, int kwidth, int kheight, float* output_data, in AveragePool() 3862 int pad_height, int kwidth, int kheight, in MaxPool() 3882 int kwidth, int kheight, float* output_data, in MaxPool()
|
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | legacy_reference_ops.h | 1491 int pad_height, int kwidth, int kheight, in AveragePool() 1566 int pad_height, int kwidth, int kheight, float* output_data, in AveragePool() 1640 int pad_height, int kwidth, int kheight, in MaxPool() 1660 int kwidth, int kheight, float* output_data, in MaxPool()
|