Home
last modified time | relevance | path

Searched defs:kwidth (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dcreate_im2col_arrays.cc42 const int kwidth = weights_shape.dims(2); in ProcessConvOperator() local
Dpropagate_fixed_sizes.cc35 void ComputeConvSizes(const Shape& input_shape, int output_depth, int kwidth, in ComputeConvSizes()
147 const int kwidth = weights_shape.dims(2); in ProcessConvOperator() local
211 const int kwidth = weights_shape.dims(2); in ProcessTransposeConvOperator() local
288 const int kwidth = weights_shape.dims(2); in ProcessDepthwiseConvOperator() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dim2col_utils.h28 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()
Dlegacy_optimized_ops.h2362 const Dims<4>& input_dims, int w, int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn()
2397 int kwidth, uint8 zero_byte, T* output_data, 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/
Dlegacy_reference_ops.h1491 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()