Home
last modified time | relevance | path

Searched refs:dilations (Results 1 – 25 of 56) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/tests/
Dconv2d_test.py48 dilations=None, argument
74 if dilations is None:
75 dilations = [1, 1]
76 dilations = [1] + dilations + [1]
87 dilations = test_utils.PermuteDimsBetweenDataFormats(
88 dilations, data_format_src, data_format_dst)
100 dilations=dilations)
140 dilations=[2, 1],
192 dilations=[2, 1],
204 dilations=[1, 2],
[all …]
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_manifest.txt341 conv/conv_channel_multiplier=1,constant_filter=True,data_format='NHWC',dilations=[1,1,1,1],filter_s…
342 conv/conv_channel_multiplier=2,constant_filter=True,data_format='NHWC',dilations=[1,1,1,1],filter_s…
343 conv/conv_channel_multiplier=1,constant_filter=True,data_format='NHWC',dilations=[1,3,2,1],filter_s…
344 conv/conv_channel_multiplier=2,constant_filter=True,data_format='NHWC',dilations=[1,3,2,1],filter_s…
345 conv/conv_channel_multiplier=1,constant_filter=True,data_format='NHWC',dilations=[1,2,2,1],filter_s…
346 conv/conv_channel_multiplier=2,constant_filter=True,data_format='NHWC',dilations=[1,2,2,1],filter_s…
347 DISABLED_conv/conv_channel_multiplier=1,constant_filter=True,data_format='NHWC',dilations=[1,1,1,1]…
348 DISABLED_conv/conv_channel_multiplier=2,constant_filter=True,data_format='NHWC',dilations=[1,1,1,1]…
349 conv/conv_channel_multiplier=1,constant_filter=True,data_format='NHWC',dilations=[1,3,2,1],filter_s…
350 conv/conv_channel_multiplier=2,constant_filter=True,data_format='NHWC',dilations=[1,3,2,1],filter_s…
[all …]
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py243 dilations, argument
298 rates=dilations,
312 dilations=None): argument
314 rates = deprecated_argument_lookup("dilations", dilations, "rates", rates)
769 dilations=None): argument
885 "dilations", dilations, "dilation_rate", dilation_rate)
892 dilations=dilation_rate,
903 dilations=None, argument
911 dilations=dilations,
927 dilations=None, argument
[all …]
Dnn_grad.py49 dilations=op.get_attr("dilations"),
57 dilations=op.get_attr("dilations"),
72 dilations=op.get_attr("dilations"),
80 dilations=op.get_attr("dilations"),
105 dilations=op.get_attr("dilations"),
112 dilations=op.get_attr("dilations"),
126 dilations=op.get_attr("dilations"),
133 dilations=op.get_attr("dilations"),
148 dilations=op.get_attr("dilations"),
156 dilations=op.get_attr("dilations"),
[all …]
Dnn_impl.py511 dilations=None): argument
554 rate = deprecated_argument_lookup("dilations", dilations, "rate", rate)
585 dilations=None, argument
632 rate=dilations,
649 dilations=None): argument
696 rate = deprecated_argument_lookup("dilations", dilations, "rate", rate)
749 dilations=None, argument
803 rate=dilations,
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py179 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, dilations, argument
204 dilations = [1] + dilations + [1]
210 dilations = test_util.NHWCToNCHW(dilations)
216 dilations=dilations,
291 dilations=full_dilation,
300 padding, dilations, rtol=1e-4): argument
305 tensor_in_sizes, filter_in_sizes, strides, dilations, padding,
324 dilations=(1, 1), argument
332 dilations = list(dilations)
340 dilations,
[all …]
Dconv_ops_3d_test.py161 dilations=full_dilation,
170 padding, dilations): argument
174 dilations[0] == 1 and dilations[1] == 1 and dilations[2] == 1)
181 tensor_in_sizes, filter_in_sizes, stride, dilations, padding,
229 dilations=[2, 1, 1])
254 dilations=[1, 2, 1])
684 strides, dilations, padding, data_format, use_gpu, argument
697 dilations[0] == 1 and dilations[1] == 1 and dilations[2] == 1)
709 full_dilations = [1] + dilations + [1]
717 dilations=full_dilations,
[all …]
Dconv2d_backprop_filter_grad_test.py84 dilations=[1, 2, 2, 1],
95 dilations=[1, 2, 2, 1],
/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.h61 const std::vector<int32>& dilations, bool is_depthwise = false)
64 dilations_(dilations), in context_()
88 virtual inline void GetDilationsInMklOrder(memory::dims* dilations) { in GetDilationsInMklOrder() argument
91 CHECK_NOTNULL(dilations); in GetDilationsInMklOrder()
95 *dilations = {dilations_rows, dilations_cols}; in GetDilationsInMklOrder()
100 *dilations = {dilations_planes, dilations_rows, dilations_cols}; in GetDilationsInMklOrder()
314 const memory::dims& strides, const memory::dims& dilations,
359 dilation_rows = dilations[0];
360 dilation_cols = dilations[1];
366 dilation_planes = dilations[0];
[all …]
Dconv_ops_3d.cc54 const std::array<int64, 3>& dilations, in launch()
62 dilations[0] == 1 && dilations[1] == 1 && dilations[2] == 1, in launch()
148 std::array<int64, 3> dilations = { in Compute() local
158 context, Get3dOutputSizeV2(input_size, filter_size, dilations, strides, in Compute()
169 dilations, strides, padding_, data_format_, in Compute()
205 const std::array<int64, 3>& dilations, in launch()
240 dilations[0] == 1 && dilations[1] == 1 && dilations[2] == 1 && in launch()
372 conv_desc.set_dilation_rate(DimIndex::X, dilations[2]) in launch()
373 .set_dilation_rate(DimIndex::Y, dilations[1]) in launch()
374 .set_dilation_rate(DimIndex::Z, dilations[0]) in launch()
[all …]
Dmkl_conv_grad_input_ops.cc59 memory::dims dilations; member
66 memory::dims dilations, memory::dims padding_left, in MklConvBwdInputParams()
72 dilations(dilations), in MklConvBwdInputParams()
190 convBwdInputDims.dilations, convBwdInputDims.padding_left, in Setup()
196 convBwdInputDims.dilations, convBwdInputDims.padding_left, in Setup()
280 key_creator.AddAsKey(convBwdInputDims.dilations); in CreateKey()
362 memory::dims padding_left, padding_right, dilations, strides; in Compute() local
370 &strides, &dilations, &fwd_output_dims_tf_order, &fwd_output_dims, in Compute()
399 for (int i = 0; i < dilations.size(); i++) dilations[i] -= 1; in Compute()
403 fwd_src_dims, fwd_filter_dims, diff_dst_dims, strides, dilations, in Compute()
Dmkl_conv_grad_filter_ops.cc57 memory::dims dilations; member
65 memory::dims dilations, memory::dims padding_left, in MklConvBwdFilterParams()
72 dilations(dilations), in MklConvBwdFilterParams()
231 convBwdFilterDims.strides, convBwdFilterDims.dilations, in Setup()
238 convBwdFilterDims.dilations, convBwdFilterDims.padding_left, in Setup()
246 convBwdFilterDims.strides, convBwdFilterDims.dilations, in Setup()
341 key_creator.AddAsKey(convBwdFilterDims.dilations); in CreateKey()
425 memory::dims padding_left, padding_right, dilations, strides, in Compute() local
434 &strides, &dilations, &fwd_dst_dims_tf_order, &fwd_dst_dims, in Compute()
464 for (int i = 0; i < dilations.size(); i++) dilations[i] -= 1; in Compute()
[all …]
Dconv_grad_ops.cc60 const gtl::ArraySlice<int32>& dilations, const std::vector<int32>& strides, in ConvBackpropExtractAndVerifyDimension() argument
67 dim->dilation = dilations[spatial_dim]; in ConvBackpropExtractAndVerifyDimension()
101 const gtl::ArraySlice<int32>& dilations, const std::vector<int32>& strides, in ConvBackpropComputeDimensionsV2() argument
152 label, input_shape, filter_shape, out_backprop_shape, dilations, in ConvBackpropComputeDimensionsV2()
Dquantized_conv_ops.cc466 std::vector<int32> dilations; in QuantizedConv2DOp() local
467 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations)); in QuantizedConv2DOp()
468 OP_REQUIRES(context, dilations.size() == 4, in QuantizedConv2DOp()
471 OP_REQUIRES(context, dilations[1] == 1 && dilations[2] == 1, in QuantizedConv2DOp()
475 OP_REQUIRES(context, (dilations[0] == 1 && dilations[3] == 1), in QuantizedConv2DOp()
Dconv_ops.cc282 TF_RETURN_IF_ERROR(context->GetAttr("dilations", &params->dilations)); in InitConv2DParameters()
295 const auto& dilations = params->dilations; in InitConv2DParameters() local
298 TF_REQUIRES(dilations.size() == 4, in InitConv2DParameters()
316 const int64 dilation_n = GetTensorDim(dilations, data_format, 'N'); in InitConv2DParameters()
317 const int64 dilation_c = GetTensorDim(dilations, data_format, 'C'); in InitConv2DParameters()
318 const int64 dilation_h = GetTensorDim(dilations, data_format, 'H'); in InitConv2DParameters()
319 const int64 dilation_w = GetTensorDim(dilations, data_format, 'W'); in InitConv2DParameters()
396 GetTensorDim(params.dilations, params.data_format, 'H'); in ComputeConv2DDimension()
398 GetTensorDim(params.dilations, params.data_format, 'W'); in ComputeConv2DDimension()
Dmkl_conv_ops.cc72 memory::dims dilations; member
84 memory::dims strides, memory::dims dilations, in MklConvFwdParams()
91 dilations(dilations), in MklConvFwdParams()
234 convFwdDims.strides, convFwdDims.dilations, convFwdDims.padding_left, in Setup()
240 convFwdDims.dilations, convFwdDims.padding_left, in Setup()
370 key_creator.AddAsKey(convFwdDims.dilations); in CreateKey()
941 dilations, strides; in Compute() local
966 &dilations, &dst_dims_tf_order, &dst_dims_mkl_order, &padding_left, in Compute()
1051 for (int i = 0; i < dilations.size(); ++i) --dilations[i]; in Compute()
1074 dst_dims_mkl_order, strides, dilations, padding_left, padding_right); in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc180 if (attrs.dilations.size() != num_dims) { in CheckConvAttrs()
184 if (attrs.dilations[batch_dim] != 1 || attrs.dilations[feature_dim] != 1) { in CheckConvAttrs()
191 if (attrs.dilations[input_dim] < 1) { in CheckConvAttrs()
194 attrs.dilations[input_dim]); in CheckConvAttrs()
205 absl::Span<const int32> dilations, const std::vector<int32>& strides, in ConvBackpropComputeDimensionsV2XlaShapes() argument
216 out_backprop_tensor_shape, dilations, strides, padding, explicit_paddings, in ConvBackpropComputeDimensionsV2XlaShapes()
228 TF_RETURN_IF_ERROR(ctx->GetAttr("dilations", &attrs.dilations)); in Create()
304 rhs_dilation[i] = attrs.dilations.at(dim); in MakeXlaForwardConvOp()
344 out_backprop_shape, attrs.dilations, attrs.strides, attrs.padding, in MakeXlaBackpropInputConvOp()
377 rhs_dilation[i] = attrs.dilations[dim]; in MakeXlaBackpropInputConvOp()
[all …]
Dconv_op_helpers.h47 std::vector<int32> dilations; member
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt69 …adding\', \'use_cudnn_on_gpu\', \'data_format\', \'name\', \'input\', \'dilations\'], varargs=None…
73 …lters\', \'output_shape\', \'strides\', \'padding\', \'data_format\', \'dilations\', \'name\'], va…
77 …r\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'data_format\', \'dilations\', \'name\', \'f…
81 …p\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'data_format\', \'dilations\', \'name\'], va…
85 …p\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'data_format\', \'dilations\', \'name\', \'f…
89 …es\', \'padding\', \'data_format\', \'name\', \'input\', \'filters\', \'dilations\'], varargs=None…
93 …argspec: "args=[\'input\', \'filter\', \'strides\', \'padding\', \'data_format\', \'dilations\', \…
97 …sizes\', \'out_backprop\', \'strides\', \'padding\', \'data_format\', \'dilations\', \'name\'], va…
101 …sizes\', \'out_backprop\', \'strides\', \'padding\', \'data_format\', \'dilations\', \'name\'], va…
105 …es\', \'padding\', \'data_format\', \'name\', \'input\', \'filters\', \'dilations\'], varargs=None…
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt69 …argspec: "args=[\'input\', \'filters\', \'stride\', \'padding\', \'data_format\', \'dilations\', \…
73 …lters\', \'output_shape\', \'strides\', \'padding\', \'data_format\', \'dilations\', \'name\'], va…
77 …argspec: "args=[\'input\', \'filters\', \'strides\', \'padding\', \'data_format\', \'dilations\', …
81 …lters\', \'output_shape\', \'strides\', \'padding\', \'data_format\', \'dilations\', \'name\'], va…
85 …argspec: "args=[\'input\', \'filters\', \'strides\', \'padding\', \'data_format\', \'dilations\', …
89 …lters\', \'output_shape\', \'strides\', \'padding\', \'data_format\', \'dilations\', \'name\'], va…
93 …lters\', \'output_shape\', \'strides\', \'padding\', \'data_format\', \'dilations\', \'name\'], va…
97 …argspec: "args=[\'input\', \'filters\', \'strides\', \'padding\', \'data_format\', \'dilations\', …
125 …argspec: "args=[\'input\', \'filter\', \'strides\', \'padding\', \'data_format\', \'dilations\', \…
129 …sizes\', \'out_backprop\', \'strides\', \'padding\', \'data_format\', \'dilations\', \'name\'], va…
[all …]
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc1532 const std::vector<int32>& dilations,
2044 std::vector<int32> dilations; in CopyAttrsConvCheckConstFilter() local
2049 TF_CHECK_OK(GetNodeAttr(orig_node->def(), "dilations", &dilations)); in CopyAttrsConvCheckConstFilter()
2061 CopyFormatAttrsConv(orig_node, nb, strides, dilations, change_format); in CopyAttrsConvCheckConstFilter()
2069 std::vector<int32> dilations; in CopyAttrsConv() local
2074 TF_CHECK_OK(GetNodeAttr(orig_node->def(), "dilations", &dilations)); in CopyAttrsConv()
2082 CopyFormatAttrsConv(orig_node, nb, strides, dilations, change_format); in CopyAttrsConv()
2094 std::vector<int32> dilations; in CopyAttrsPadWithConv2D() local
2100 TF_CHECK_OK(GetNodeAttr(orig_node->def(), "dilations", &dilations)); in CopyAttrsPadWithConv2D()
2113 nb->Attr("dilations", dilations); in CopyAttrsPadWithConv2D()
[all …]
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dconv2d_test.py49 dilations = [1] + list(dilation_rate) + [1]
53 dilations = [1, 1] + list(dilation_rate)
60 dilations=dilations,
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc662 const auto& dilations = GetListAttr(node, "dilations"); in ConvertConvOperator() local
664 ExpectValue(dilations.i_size(), 4, "number of dilations")); in ConvertConvOperator()
665 if (dilations.i(0) != 1 || dilations.i(3) != 1) { in ConvertConvOperator()
669 node.name(), "\" had dilations:[ ", dilations.i(0), ", ", in ConvertConvOperator()
670 dilations.i(1), ", ", dilations.i(2), ", ", dilations.i(3), "].")); in ConvertConvOperator()
672 conv->dilation_height_factor = dilations.i(1); in ConvertConvOperator()
673 conv->dilation_width_factor = dilations.i(2); in ConvertConvOperator()
735 const auto& dilations = GetListAttr(node, "dilations"); in ConvertDepthwiseConvOperator() local
737 ExpectValue(dilations.i_size(), 4, "number of dilations")); in ConvertDepthwiseConvOperator()
738 if (dilations.i(0) != 1 || dilations.i(3) != 1) { in ConvertDepthwiseConvOperator()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc121 const std::array<int64, 3>& dilations, in Get3dOutputSizeV2() argument
127 input[i], window[i], dilations[i], strides[i], padding_type, in Get3dOutputSizeV2()
444 std::vector<int32> dilations; in Conv2DShapeImpl() local
445 TF_RETURN_IF_ERROR(c->GetAttr("dilations", &dilations)); in Conv2DShapeImpl()
447 if (dilations.size() != 4) { in Conv2DShapeImpl()
450 dilations.size()); in Conv2DShapeImpl()
466 const int32 dilation_rows = GetTensorDim(dilations, data_format, 'H'); in Conv2DShapeImpl()
467 const int32 dilation_cols = GetTensorDim(dilations, data_format, 'W'); in Conv2DShapeImpl()
565 std::vector<int32> dilations; in Conv3DShape() local
566 TF_RETURN_IF_ERROR(c->GetAttr("dilations", &dilations)); in Conv3DShape()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc471 string dilations; in ToString() local
475 port::Appendf(&dilations, "%lld ", this->dilations()[i]); in ToString()
482 strides.c_str(), dilations.c_str()); in ToString()
495 port::Appendf(&desc, "_d%d:%lld", i, dilations()[i]); in ToShortString()

123