| /external/tensorflow/tensorflow/compiler/tests/ |
| conv2d_test.py | 41 dilations=None,
52 dilations: RHS dilations.
61 if dilations is None:
62 dilations = [1, 1]
63 dilations = [1] + dilations + [1]
75 dilations=dilations)
107 dilations=[2, 1]
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| conv_ops_test.py | 168 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, dilations,
177 dilations: Dilated rate: [col_dilation, row_dilation]
201 dilations = [1] + dilations + [1]
205 dilations = test_util.NHWCToNCHW(dilations)
209 dilations=dilations,
293 dilations=full_dilation,
302 padding, dilations)
[all...] |
| conv2d_backprop_filter_grad_test.py | 82 dilations=[1, 2, 2, 1],
93 dilations=[1, 2, 2, 1],
|
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_grad_ops.cc | 50 const gtl::ArraySlice<int32>& dilations, const std::vector<int32>& strides,
57 dim->dilation = dilations[spatial_dim];
100 const gtl::ArraySlice<int32>& dilations, const std::vector<int32>& strides,
144 label, input_shape, filter_shape, out_backprop_shape, dilations,
|
| quantized_conv_ops.cc | 466 std::vector<int32> dilations; local
467 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations));
468 OP_REQUIRES(context, dilations.size() == 4,
469 errors::InvalidArgument("Dilations field must "
471 OP_REQUIRES(context, dilations[1] == 1 && dilations[2] == 1,
475 OP_REQUIRES(context, (dilations[0] == 1 && dilations[3] == 1),
478 "dilations in the batch and depth dimensions."))
[all...] |
| conv_grad_ops.h | 255 const gtl::ArraySlice<int32>& dilations, const std::vector<int32>& strides,
|
| conv_grad_filter_ops.cc | 202 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_));
204 errors::InvalidArgument("Sliding window dilations field must "
209 "dilations in the batch and depth dimensions."));
315 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_));
317 errors::InvalidArgument("Sliding window dilations field must "
322 "dilations in the batch and depth dimensions."));
557 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_));
559 errors::InvalidArgument("Sliding window dilations field must "
568 "dilations in the batch and depth dimensions."));
640 std::vector<int32> dilations(4, 1)
[all...] |
| conv_grad_input_ops.cc | 205 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_));
207 errors::InvalidArgument("Sliding window dilations field must "
212 "dilations in the batch and depth dimensions."));
320 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_));
322 errors::InvalidArgument("Sliding window dilations field must "
327 "dilations in the batch and depth dimensions."));
632 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_));
634 errors::InvalidArgument("Sliding window dilations field must "
643 "dilations in the batch and depth dimensions."));
710 std::vector<int32> dilations(4, 1)
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| nn_grad.py | 50 dilations=op.get_attr("dilations"),
58 dilations=op.get_attr("dilations"),
73 dilations=op.get_attr("dilations"),
81 dilations=op.get_attr("dilations"),
499 dilations = op.get_attr("dilations")
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.cc | 439 string dilations; local
443 port::Appendf(&dilations, "%lld ", dilation_rates_[i]);
450 strides.c_str(), dilations.c_str());
|
| dnn.h | 572 std::vector<int64> dilations() const { return dilation_rates_; } function in class:perftools::gputools::dnn::ConvolutionDescriptor
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| common_shape_fns.cc | 105 const std::array<int64, 3>& dilations,
111 input[i], window[i], dilations[i], strides[i], padding_type,
404 std::vector<int32> dilations; local
405 TF_RETURN_IF_ERROR(c->GetAttr("dilations", &dilations));
407 if (dilations.size() != 4) {
410 dilations.size());
426 const int32 dilation_rows = GetTensorDim(dilations, data_format, 'H');
427 const int32 dilation_cols = GetTensorDim(dilations, data_format, 'W');
[all...] |
| common_shape_fns.h | 136 const std::array<int64, 3>& dilations,
|
| common_shape_fns_test.cc | 547 auto set_op = [&op](const std::vector<int32>& dilations,
553 .Attr("dilations", dilations)
569 // 1x1 filter, 2x1 dilations, 1x1 strides
573 // 1x1 filter, 2x1 dilations, 2x1 strides
577 // 1x1 filter, 2x1 dilations, 2x2 strides
581 // 3x3 filter, 2x1 dilations, 1x1 strides
585 // 3x3 filter, 2x1 dilations, 2x1 strides
589 // 3x3 filter, 1x2 dilations, 2x2 strides
594 // 1x1 filter, 2x1 dilations, 1x1 stride
[all...] |
| /external/opencv/cv/src/ |
| cvcalibinit.cpp | 229 int quad_count = 0, group_idx = 0, i = 0, dilations = 0; local
272 // Try our standard "1" dilation, but if the pattern is not found, iterate the whole procedure with higher dilations.
274 // we want to use the minimum number of dilations possible since dilations cause the squares to become smaller,
278 for( dilations = min_dilations; dilations <= max_dilations; dilations++ )
287 CV_CALL( quad_count = icvGenerateQuadsEx( &quads, &corners, storage, img, thresh_img, dilations, flags ));
300 if (dilations > 0)
301 cvDilate( thresh_img, thresh_img, 0, dilations-1 )
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 624 const auto& dilations64 = convolution_descriptor.dilations();
633 std::vector<int> dilations(convolution_descriptor.ndims());
640 std::transform(dilations64.cbegin(), dilations64.cend(), dilations.begin(),
645 strides.data(), dilations.data(),
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
