Home
last modified time | relevance | path

Searched refs:filter_height (Results 1 – 25 of 69) sorted by relevance

123

/external/tensorflow/tensorflow/lite/kernels/
Dpooling.cc95 compute_out_size(height, params->filter_height, params->stride_height); in GenericPrepare()
98 params->filter_height, out_height); in GenericPrepare()
133 op_params.filter_height = params->filter_height; \ in AverageEvalFloat()
163 op_params.filter_height = params->filter_height; \ in AverageEvalQuantizedUint8()
190 op_params.filter_height = params->filter_height; in AverageEvalQuantizedInt8()
212 op_params.filter_height = params->filter_height; \ in MaxEvalFloat()
240 op_params.filter_height = params->filter_height; \ in MaxEvalQuantizedUInt8()
269 op_params.filter_height = params->filter_height; \ in MaxEvalQuantizedInt8()
293 op_params.filter_height = params->filter_height; \ in L2EvalFloat()
Dpadding.h44 int in_width, int filter_height, int filter_width, TfLitePadding padding) { in ComputePaddingHeightWidth() argument
47 ComputeOutSize(padding, in_height, filter_height, stride_height); in ComputePaddingHeightWidth()
51 ComputePadding(stride_height, 1, in_height, filter_height, out_height); in ComputePaddingHeightWidth()
Dtranspose_conv.cc123 const int filter_height = SizeOfDimension(weights, 1); in ResizeIm2ColTensor() local
124 im2col_shape_array->data[3] = input_depth * filter_height * filter_width; in ResizeIm2ColTensor()
201 const int filter_height = SizeOfDimension(weights, 1); in Eval() local
208 filter_height, filter_width, params->padding); in Eval()
Dconv.cc152 int filter_height = filter->dims->data[1]; in AllocateTemporaryTensorsIfRequired() local
161 filter_height != 1); in AllocateTemporaryTensorsIfRequired()
275 int filter_height = filter->dims->data[1]; in Prepare() local
293 compute_out_size(height, filter_height, params->stride_height, in Prepare()
298 height, filter_height, out_height); in Prepare()
345 im2col_size->data[3] = input_depth * filter_height * filter_width; in Prepare()
367 hwcn_weights_size->data[0] = (filter_height * filter_width * input_depth); in Prepare()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_using_gemm.cc90 int input_depth, const T2* filter_data, int filter_height, in operator ()() argument
114 filter_top_offset = ((output_height - 1) * stride_rows + filter_height - in operator ()()
121 ((output_height - 1) * stride_rows + filter_height - input_height) / in operator ()()
155 for (int filter_y = 0; filter_y < filter_height; ++filter_y) { in operator ()()
213 int input_depth, const T2* filter_data, int filter_height, in operator ()() argument
224 if ((filter_width <= 0) || (filter_height <= 0) || (filter_count <= 0)) { in operator ()()
226 << filter_width << ", " << filter_height << ", " in operator ()()
238 if (filter_height == 1 && filter_width == 1 && stride_rows == 1 && in operator ()()
251 } else if (filter_height == input_height && filter_width == input_width && in operator ()()
275 filter_top_offset = ((output_height - 1) * stride_rows + filter_height - in operator ()()
[all …]
Deigen_benchmark_cpu_test.cc35 int filter_count, int filter_height, int filter_width) { in SpatialConvolution() argument
46 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolution()
54 num_computed_elements * (input_depth * filter_height * filter_width); in SpatialConvolution()
63 int filter_count, int filter_height, in SpatialConvolutionBackwardInput() argument
75 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardInput()
82 num_computed_elements * (input_depth * filter_height * filter_width); in SpatialConvolutionBackwardInput()
91 int filter_count, int filter_height, in SpatialConvolutionBackwardKernel() argument
103 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardKernel()
256 int filter_count, int filter_height, int filter_width, in CuboidConvolution() argument
269 filter_height, filter_width, filter_planes, input_depth, filter_count); in CuboidConvolution()
[all …]
Ddepthwise_conv_op_gpu.h81 const int filter_height = in DepthwiseConv2dGPUKernelNHWC() local
108 const int input_row_end = input_row_start + filter_height; in DepthwiseConv2dGPUKernelNHWC()
116 UNROLL for (int filter_row = 0; filter_row < filter_height; in DepthwiseConv2dGPUKernelNHWC()
135 UNROLL for (int filter_row = 0; filter_row < filter_height; in DepthwiseConv2dGPUKernelNHWC()
188 const int filter_height = in DepthwiseConv2dGPUKernelNHWCSmall() local
205 const int filter_pixels = filter_height * filter_width; in DepthwiseConv2dGPUKernelNHWCSmall()
208 const int tile_height = in_height + filter_height - even_height; in DepthwiseConv2dGPUKernelNHWCSmall()
283 UNROLL for (int r = 0; r < filter_height; ++r) { in DepthwiseConv2dGPUKernelNHWCSmall()
321 const int filter_height = in DepthwiseConv2dGPUKernelNCHW() local
388 const int input_row_end = input_row_start + filter_height; in DepthwiseConv2dGPUKernelNCHW()
[all …]
Dquantized_conv_ops.cc57 int filter_height, int filter_width, int filter_count, in operator ()() argument
93 ((output_height - 1) * stride + filter_height - input_height + 1) / 2; in operator ()()
98 ((output_height - 1) * stride + filter_height - input_height) / 2; in operator ()()
132 for (int filter_y = 0; filter_y < filter_height; ++filter_y) { in operator ()()
203 int filter_height, int filter_width, int filter_count, in operator ()() argument
224 filter_height, filter_width, filter_count, filter_offset, in operator ()()
238 ((output_height - 1) * stride + filter_height - input_height + 1) / 2; in operator ()()
243 ((output_height - 1) * stride + filter_height - input_height) / 2; in operator ()()
257 const int filter_value_count = filter_width * filter_height * input_depth; in operator ()()
312 for (int filter_y = 0; filter_y < filter_height; ++filter_y) { in operator ()()
Dconv_ops_fused_image_transform.cc262 const T2* filter_data, int filter_height, int filter_width, in operator ()() argument
275 if ((filter_width <= 0) || (filter_height <= 0) || (filter_count <= 0)) { in operator ()()
277 << filter_width << ", " << filter_height << ", " in operator ()()
299 filter_top_offset = ((output_height - 1) * stride_rows + filter_height - in operator ()()
306 ((output_height - 1) * stride_rows + filter_height - padded_height) / in operator ()()
332 const int filter_value_count = filter_width * filter_height * input_depth; in operator ()()
375 filter_height * task_params.cache_line_width * input_depth; in operator ()()
431 ((output_height * stride_rows) - filter_top_offset) + filter_height; in operator ()()
438 if (end_cached_lines < (in_y_origin + filter_height)) { in operator ()()
548 for (int filter_y = 0; filter_y < filter_height; ++filter_y) { in operator ()()
Dops_util_test.cc35 int filter_height; member
73 pad_struct.input.in_height, pad_struct.input.filter_height, in VerifyGet2dOutputSizeBoundaries()
88 pad_struct.input.in_height, pad_struct.input.filter_height, in VerifyGet2dOutputSizeValues()
107 pad_struct.input.in_height, pad_struct.input.filter_height, in VerifyGet2dOutputVerboseSizeValues()
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc55 int height, int filter_width, int filter_height, in CalculateOpData() argument
59 filter_height, out_height); in CalculateOpData()
123 const int filter_height = filter_shape.Dims(1); in DepthwiseConvOptimizedForFilterWidthEight() local
132 output_depth * filter_width * filter_height * input_depth; in DepthwiseConvOptimizedForFilterWidthEight()
143 {1, output_depth, filter_height, filter_width}); in DepthwiseConvOptimizedForFilterWidthEight()
149 for (int filter_y = 0; filter_y < filter_height; ++filter_y) { in DepthwiseConvOptimizedForFilterWidthEight()
179 int filter_y_end = filter_height; in DepthwiseConvOptimizedForFilterWidthEight()
180 if ((in_y_origin + filter_height) >= input_height) { in DepthwiseConvOptimizedForFilterWidthEight()
181 filter_y_end -= (in_y_origin + filter_height) - input_height; in DepthwiseConvOptimizedForFilterWidthEight()
344 const int filter_height = GetTensorShape(filter).Dims(1); in EvalQuantized() local
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dmultithreaded_conv.h90 int filter_height, int filter_width, int filter_count, in operator()
94 const bool is_1x1_kernel = (filter_height == 1 && filter_width == 1 && in operator()
108 } else if (filter_height == input_height && filter_width == input_width && in operator()
113 filter_width * filter_height * input_depth; in operator()
126 ConstEigenTensor filter(filter_data, filter_height, filter_width, in operator()
158 const int filter_height = filter_shape.Dims(1); in Conv() local
164 input_width, input_depth, filter_data, filter_height, in Conv()
Dlegacy_optimized_ops.h1211 params.filter_height = kheight; in AveragePool()
1239 int filter_height, float* output_data, in AveragePool() argument
1242 filter_width, filter_height, output_data, output_dims); in AveragePool()
1247 int pad_height, int filter_width, int filter_height, in AveragePool() argument
1254 params.filter_height = filter_height; in AveragePool()
1268 int pad_height, int filter_width, int filter_height, in AveragePool() argument
1281 pad_height, filter_width, filter_height, output_activation_min, in AveragePool()
1289 int filter_height, int32 output_activation_min, in AveragePool() argument
1293 filter_width, filter_height, output_activation_min, in AveragePool()
1305 params.filter_height = kheight; in MaxPool()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_float_test.cc76 const int filter_height = ExponentialRandomPositiveInt(0.9f, 4, 10); in TryTestOneDepthwiseConv() local
107 filter_height, stride, dilation_width_factor, in TryTestOneDepthwiseConv()
113 {1, filter_height, filter_width, output_depth}); in TryTestOneDepthwiseConv()
123 filter_width * filter_height * input_amplitude * filter_amplitude; in TryTestOneDepthwiseConv()
Ddepthwiseconv_quantized_test.cc491 int filter_width, int filter_height, in TryTestDepthwiseConv() argument
525 filter_height, stride, dilation_width_factor, in TryTestDepthwiseConv()
531 {1, filter_height, filter_width, output_depth}); in TryTestDepthwiseConv()
566 const int filter_height = ExponentialRandomPositiveInt(0.9f, 4, 10); in TryTestOneDepthwiseConv() local
576 input_height, filter_width, filter_height, depth_multiplier, stride, in TryTestOneDepthwiseConv()
588 const int filter_height = 3; in TryTestOneDepthwiseConv3x3Filter() local
618 input_height, filter_width, filter_height, depth_multiplier, stride, in TryTestOneDepthwiseConv3x3Filter()
640 const int filter_height = 3; in TryTestOneNeonDot3x3() local
654 input_height, filter_width, filter_height, depth_multiplier, stride, in TryTestOneNeonDot3x3()
Dtest_util.cc24 int filter_width, int filter_height, int stride, in ComputeConvSizes() argument
33 int dilated_filter_height = dilation_height_factor * (filter_height - 1) + 1; in ComputeConvSizes()
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Ddepthwise_conv.cc52 int height, int filter_width, int filter_height, in CalculateOpData() argument
56 filter_height, out_height); in CalculateOpData()
170 int filter_height = SizeOfDimension(filter, 1); in Eval() local
173 int out_height = ComputeOutSize(params->padding, height, filter_height, in Eval()
178 filter_width, filter_height, out_width, in Eval()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dflatten_atrous.cc69 const int32 filter_height = filter.dim_size(0); in FlattenAtrousConv() local
75 (filter_height - 1) * block_height + 1; in FlattenAtrousConv()
87 for (int h = 0; h < filter_height; ++h) { in FlattenAtrousConv()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Conv2D.pbtxt14 `[filter_height, filter_width, in_channels, out_channels]`
71 `[filter_height, filter_width, in_channels, out_channels]`, this op
75 `[filter_height * filter_width * in_channels, output_channels]`.
78 filter_height * filter_width * in_channels]`.
Dapi_def_Dilation2DBackpropFilter.pbtxt12 3-D with shape `[filter_height, filter_width, depth]`.
24 3-D with shape `[filter_height, filter_width, depth]`.
Dapi_def_Dilation2D.pbtxt12 3-D with shape `[filter_height, filter_width, depth]`.
44 `filter` tensor has shape `[filter_height, filter_width, depth]`, i.e., each
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h1407 params.filter_height = kheight; in AveragePool()
1489 int filter_height, float* output_data, in AveragePool() argument
1492 filter_width, filter_height, output_data, output_dims); in AveragePool()
1497 int pad_height, int filter_width, int filter_height, in AveragePool() argument
1504 params.filter_height = filter_height; in AveragePool()
1518 int pad_height, int filter_width, int filter_height, in AveragePool() argument
1531 pad_height, filter_width, filter_height, output_activation_min, in AveragePool()
1539 int filter_height, int32 output_activation_min, in AveragePool() argument
1543 filter_width, filter_height, output_activation_min, in AveragePool()
1555 params.filter_height = kheight; in MaxPool()
[all …]
Ddepthwiseconv_float.h49 const int filter_height = filter_shape.Dims(1); in DepthwiseConv() local
65 for (int filter_y = 0; filter_y < filter_height; ++filter_y) { in DepthwiseConv()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dconv.h60 const int filter_height = filter_shape.Dims(1); in ConvPerChannel() local
71 for (int filter_y = 0; filter_y < filter_height; ++filter_y) { in ConvPerChannel()
Ddepthwise_conv.h55 const int filter_height = filter_shape.Dims(1); in DepthwiseConvPerChannel() local
71 for (int filter_y = 0; filter_y < filter_height; ++filter_y) { in DepthwiseConvPerChannel()

123