Home
last modified time | relevance | path

Searched refs:stride_width (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/lite/kernels/
Dconv_test.cc47 const TensorData& filter, const TensorData& output, int stride_width = 2, in BaseConvolutionOpModel() argument
74 CreateConv2DOptions(builder_, padding, stride_width, in BaseConvolutionOpModel()
178 const int stride_width = 1; in TEST_P() local
186 {TensorType_FLOAT32, {}}, stride_width, stride_height, padding); in TEST_P()
232 const int stride_width = 1; in TEST_P() local
240 {TensorType_FLOAT32, {}}, stride_width, stride_height, padding); in TEST_P()
286 const int stride_width = 1; in TEST_P() local
294 {TensorType_FLOAT32, {}}, stride_width, stride_height, padding, in TEST_P()
342 const int stride_width = 1; in TEST_P() local
350 {TensorType_FLOAT32, {}}, stride_width, stride_height, padding); in TEST_P()
Dconv.cc170 int outWidth = computeOutSize(width, filter_width, params->stride_width); in Prepare()
176 ComputePadding(params->stride_width, width, filter_width, outWidth); in Prepare()
206 (params->stride_width != 1 || params->stride_height != 1 || in Prepare()
308 params->stride_width, params->stride_height, data->padding.width, in EvalQuantized()
323 params->stride_width, params->stride_height, data->padding.width, in EvalQuantized()
347 params->stride_width, params->stride_height, in EvalFloat()
358 params->stride_width, params->stride_height, in EvalFloat()
375 GetTensorDims(bias), params->stride_width, params->stride_height, in EvalFloat()
386 params->stride_width, params->stride_height, in EvalFloat()
Dpooling.cc93 computeOutSize(width, params->filter_width, params->stride_width); in GenericPrepare()
99 data->padding.width = ComputePadding(params->stride_width, width, in GenericPrepare()
131 GetTensorData<float>(input), GetTensorDims(input), params->stride_width, \ in AverageEvalFloat()
153 params->stride_width, params->stride_height, \ in AverageEvalQuantized()
175 GetTensorData<float>(input), GetTensorDims(input), params->stride_width, \ in MaxEvalFloat()
197 params->stride_width, params->stride_height, \ in MaxEvalQuantized()
219 GetTensorData<float>(input), GetTensorDims(input), params->stride_width, \ in L2EvalFloat()
Ddepthwise_conv.cc139 int out_width = compute_out_size(width, filter_width, params->stride_width); in Prepare()
146 ComputePadding(params->stride_width, width, filter_width, out_width); in Prepare()
190 GetTensorData<float>(bias), GetTensorDims(bias), params->stride_width, in EvalFloat()
218 GetTensorData<int32_t>(bias), GetTensorDims(bias), params->stride_width, in EvalQuantized()
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
Doperator_test.cc186 op.stride_width = 123; in TEST_F()
193 EXPECT_EQ(op.stride_width, output_toco_op->stride_width); in TEST_F()
218 op.stride_width = 123; in TEST_F()
225 EXPECT_EQ(op.stride_width, output_toco_op->stride_width); in TEST_F()
267 op.stride_width = 123; in TEST_F()
276 EXPECT_EQ(op.stride_width, output_toco_op->stride_width); in TEST_F()
285 op.stride_width = 123; in TEST_F()
291 EXPECT_EQ(op.stride_width, output_toco_op->stride_width); in TEST_F()
300 op.stride_width = 123; in TEST_F()
307 EXPECT_EQ(op.stride_width, output_toco_op->stride_width); in TEST_F()
Doperator.cc41 return ::tflite::CreatePool2DOptions(*builder, padding, op.stride_width, in WriteOptions()
49 op->stride_width = options.stride_w(); in ReadOptions()
70 return ::tflite::CreateConv2DOptions(*builder, padding, op.stride_width, in WriteOptions()
77 op->stride_width = options.stride_w(); in ReadOptions()
98 *builder, padding, op.stride_width, op.stride_height, in WriteOptions()
105 op->stride_width = options.stride_w(); in ReadOptions()
354 return ::tflite::CreatePool2DOptions(*builder, padding, op.stride_width, in WriteOptions()
362 op->stride_width = options.stride_w(); in ReadOptions()
404 return ::tflite::CreatePool2DOptions(*builder, padding, op.stride_width, in WriteOptions()
412 op->stride_width = options.stride_w(); in ReadOptions()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
Ddepthwiseconv_float.h28 int stride_width, int stride_height, int pad_width, in DepthwiseConv() argument
50 const int in_x_origin = (out_x * stride_width) - pad_width; in DepthwiseConv()
89 int stride_width, int stride_height, int pad_width, in DepthwiseConv() argument
95 bias_dims, stride_width, stride_height, pad_width, pad_height, in DepthwiseConv()
Ddepthwiseconv_uint8.h33 int stride_width, int stride_height, int pad_width, in DepthwiseConv() argument
56 const int in_x_origin = (out_x * stride_width) - pad_width; in DepthwiseConv()
99 int stride_width, int stride_height, int pad_width, in DepthwiseConv() argument
109 filter_offset, bias_data, bias_dims, stride_width, in DepthwiseConv()
Dreference_ops.h160 int stride_width, int stride_height, int pad_width, in Conv() argument
183 const int in_x_origin = (out_x * stride_width) - pad_width; in Conv()
223 const float* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv() argument
230 stride_width, stride_height, pad_width, pad_height, in Conv()
252 int stride_width, int stride_height, int pad_width, in Conv() argument
277 const int in_x_origin = (out_x * stride_width) - pad_width; in Conv()
322 int stride_width, int stride_height, int pad_width, in Conv() argument
339 filter_offset, bias_data, bias_dims, stride_width, stride_height, in Conv()
1686 int stride_width, int stride_height, int pad_width, in AveragePool() argument
1701 const int in_x_origin = (out_x * stride_width) - pad_width; in AveragePool()
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Dcblas_conv.h37 int stride_width, int stride_height, int pad_width, in Conv() argument
48 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv()
52 optimized_ops::Im2col(input_data, input_dims, stride_width, stride_height, in Conv()
Doptimized_ops.h638 int stride_width, int stride_height, int pad_width, int pad_height, in ExtractPatchIntoBufferColumn() argument
649 const int iw_ungated_start = w * stride_width - pad_width; in ExtractPatchIntoBufferColumn()
722 void Im2col(const T* input_data, const Dims<4>& input_dims, int stride_width, in Im2col() argument
743 input_dims, w, h, b, kheight, kwidth, stride_width, stride_height, in Im2col()
764 int stride_width, int stride_height, int pad_width, in Conv() argument
777 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv()
781 Im2col(input_data, input_dims, stride_width, stride_height, pad_width, in Conv()
812 const float* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv() argument
819 stride_width, stride_height, pad_width, pad_height, in Conv()
841 int stride_width, int stride_height, int pad_width, in Conv() argument
[all …]
Dmultithreaded_conv.h167 int stride_width, int stride_height, int pad_width, in Conv() argument
184 output_depth, stride_height, stride_width, pad_height, pad_width, in Conv()
Ddepthwiseconv_float.h913 int stride_width, int stride_height, int pad_width,
947 if (!row_accum_func && (stride_width == 1 || ALLOW_STRIDED) && \
1017 row_accum_func(stride_width, input_depth, input_width,
1075 int stride_width, int stride_height, int pad_width,
1081 bias_dims, stride_width, stride_height, pad_width, pad_height,
/external/tensorflow/tensorflow/contrib/lite/
Dbuiltin_op_data.h52 int stride_width; member
59 int stride_width; member
71 int stride_width; member
Dnnapi_delegate.cc181 add_scalar_int32(builtin->stride_width); in AddOpsAndParams()
191 add_scalar_int32(builtin->stride_width); in AddOpsAndParams()
199 add_scalar_int32(builtin->stride_width); in AddOpsAndParams()
Dmodel.cc267 params->stride_width = conv_params->stride_w(); in ParseOpData()
299 params->stride_width = pool_params->stride_w(); in ParseOpData()
314 params->stride_width = conv_params->stride_w(); in ParseOpData()
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dconvert_pure_conv_to_depthwise.cc33 if (conv_op->stride_width != conv_op->stride_height) { in Run()
69 depthwiseconv_op->stride_width = conv_op->stride_width; in Run()
Dcreate_im2col_arrays.cc48 if (kwidth == 1 && kheight == 1 && conv_op->stride_width == 1 && in Run()
63 LogName(*conv_op), kwidth, kheight, conv_op->stride_width, in Run()
Didentify_l2_pool.cc87 l2pool_op->stride_width = avpool_op->stride_width; in Run()
Dpropagate_fixed_sizes.cc33 int kheight, int stride_width, int stride_height, in ComputeConvSizes() argument
44 output_width = (input_width + stride_width - kwidth) / stride_width; in ComputeConvSizes()
47 output_width = (input_width + stride_width - 1) / stride_width; in ComputeConvSizes()
55 0, ((output_width - 1) * stride_width + kwidth - input_width) / 2); in ComputeConvSizes()
168 ComputeConvSizes(input_shape, output_depth, kwidth, kheight, op->stride_width, in ProcessConvOperator()
224 ComputeConvSizes(input_shape, output_depth, kwidth, kheight, op->stride_width, in ProcessDepthwiseConvOperator()
700 op->stride_width, op->stride_height, op->padding.type, in ProcessAveragePoolOperator()
717 op->stride_width, op->stride_height, op->padding.type, in ProcessMaxPoolOperator()
736 op->stride_width, op->stride_height, op->padding.type, in ProcessL2PoolOperator()
/external/tensorflow/tensorflow/contrib/lite/toco/
Dmodel.h357 int stride_width = 0;
378 int stride_width = 0;
638 int stride_width = 0;
669 int stride_width = 0;
684 int stride_width = 0;
853 int stride_width = 0;
Ddump_graphviz.cc247 AppendF(&node_properties.label, "\\n%dx%d/%s", conv_op.stride_width, in GetPropertiesForOperator()
255 AppendF(&node_properties.label, "\\n%dx%d/%s", conv_op.stride_width, in GetPropertiesForOperator()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Dilation2DBackpropInput.pbtxt31 the input tensor. Must be: `[1, stride_height, stride_width, 1]`.
Dapi_def_Dilation2DBackpropFilter.pbtxt31 the input tensor. Must be: `[1, stride_height, stride_width, 1]`.
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_test.cc420 const int stride_width = 3; in AnisotropicStrides() local
426 .Attr("strides", {1, stride_height, stride_width, 1}) in AnisotropicStrides()

12