Searched defs:kwidth (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | create_im2col_arrays.cc | 42 const int kwidth = weights_shape.dims(2); in ProcessConvOperator() local
|
| D | propagate_fixed_sizes.cc | 34 void ComputeConvSizes(const Shape& input_shape, int output_depth, int kwidth, in ComputeConvSizes()
146 const int kwidth = weights_shape.dims(2); in ProcessConvOperator() local
210 const int kwidth = weights_shape.dims(2); in ProcessTransposeConvOperator() local
287 const int kwidth = weights_shape.dims(2); in ProcessDepthwiseConvOperator() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 449 const Dims<4>& input_dims, int w, int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn()
484 int kwidth, uint8 zero_byte, T* output_data, in Im2col()
503 int pad_width, int pad_height, int kheight, int kwidth, in Im2col()
711 int pad_width, int pad_height, int kheight, int kwidth, in Im2col()
1204 int pad_height, int kwidth, int kheight, in AveragePool()
1225 int pad_height, int kwidth, int kheight, float* output_data, in AveragePool()
1299 int pad_height, int kwidth, int kheight, in MaxPool()
1319 int kwidth, int kheight, float* output_data, in MaxPool()
|
| D | optimized_ops.h | 1974 int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn()
2146 void Im2col(const ConvParams& params, int kheight, int kwidth, uint8 zero_byte, in Im2col()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 1400 int pad_height, int kwidth, int kheight, in AveragePool()
1475 int pad_height, int kwidth, int kheight, float* output_data, in AveragePool()
1549 int pad_height, int kwidth, int kheight, in MaxPool()
1569 int kwidth, int kheight, float* output_data, in MaxPool()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1837 int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn()
1929 const Dims<4>& input_dims, int w, int h, int b, int kheight, int kwidth, in ExtractPatchIntoBufferColumn()
2047 void Im2col(const ConvParams& params, int kheight, int kwidth, uint8 zero_byte, in Im2col()
2086 int kwidth, uint8 zero_byte, T* output_data, in Im2col()
2105 int pad_width, int pad_height, int kheight, int kwidth, in Im2col()
2565 int pad_width, int pad_height, int kheight, int kwidth, in Im2col()
|