| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | depthwise_conv_op_test.py | 74 def _DepthwiseConv2dNumpy(x1, x2, strides, padding, data_format, dilations): argument
98 if dilations:
99 dilations = [dilations[0], dilations[3], dilations[1], dilations[2]]
101 if dilations:
105 new_fh = (fh - 1) * dilations[0] + 1
106 new_fw = (fw - 1) * dilations[1] + 1
110 new_x2[i * dilations[0], j * dilations[1], : :] = x2[i, j, :, :]
149 dilations=None): argument
150 return input_size, filter_size, out_size, stride, padding, dilations
163 dilations=[2, 2]),
[all …]
|
| D | conv_ops_test.py | 183 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, dilations, argument
208 dilations = [1] + dilations + [1]
214 dilations = test_util.NHWCToNCHW(dilations)
220 dilations=dilations,
296 dilations=full_dilation,
305 padding, dilations, rtol=1e-4): argument
310 tensor_in_sizes, filter_in_sizes, strides, dilations, padding,
329 dilations=(1, 1), argument
337 dilations = list(dilations)
348 dilations,
[all …]
|
| D | conv_ops_3d_test.py | 161 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,
295 dilations=[2, 1, 1])
323 dilations=[1, 2, 1])
405 dilations=[1, 1, 1, 1, 1],
780 strides, dilations, padding, data_format, use_gpu, argument
793 dilations[0] == 1 and dilations[1] == 1 and dilations[2] == 1)
805 full_dilations = [1] + dilations + [1]
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | conv2d_test.py | 48 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 …]
|
| D | conv_node_name_test.py | 42 dilations): argument
58 dilation_rate=dilations,
67 dilation_rate=dilations,
97 dilations = [2]
100 dilations)
106 dilations = [2, 2]
109 dilations)
115 dilations = [2, 2, 2]
118 dilations)
|
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | test_manifest.txt | 341 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/ |
| D | nn_ops.py | 334 dilations, argument
389 rates=dilations,
404 dilations=None): argument
406 rates = deprecated_argument_lookup("dilations", dilations, "rates", rates)
875 dilations=None): # pylint: disable=g-doc-args argument
991 "dilations", dilations, "dilation_rate", dilation_rate)
998 dilations=dilation_rate,
1010 dilations=None, argument
1018 dilations=dilations,
1034 dilations=None, argument
[all …]
|
| D | nn_grad.py | 49 dilations=op.get_attr("dilations"),
58 dilations=op.get_attr("dilations"),
76 dilations=op.get_attr("dilations"),
85 dilations=op.get_attr("dilations"),
111 dilations=op.get_attr("dilations"),
119 dilations=op.get_attr("dilations"),
134 dilations=op.get_attr("dilations"),
142 dilations=op.get_attr("dilations"),
158 dilations=op.get_attr("dilations"),
166 dilations=op.get_attr("dilations"),
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 67 const std::vector<int32>& dilations, bool is_depthwise = false)
70 dilations_(dilations), in context_()
94 virtual inline void GetDilationsInMklOrder(memory::dims* dilations) { in GetDilationsInMklOrder() argument
97 DCHECK(dilations); in GetDilationsInMklOrder()
101 *dilations = {dilations_rows, dilations_cols}; in GetDilationsInMklOrder()
106 *dilations = {dilations_planes, dilations_rows, dilations_cols}; in GetDilationsInMklOrder()
320 const memory::dims& strides, const memory::dims& dilations,
365 dilation_rows = dilations[0];
366 dilation_cols = dilations[1];
372 dilation_planes = dilations[0];
[all …]
|
| D | mkl_conv_grad_input_ops.cc | 67 memory::dims dilations; member
74 memory::dims dilations, memory::dims padding_left, in MklConvBwdInputParams()
82 dilations(dilations), in MklConvBwdInputParams()
204 convBwdInputDims.dilations, convBwdInputDims.padding_left, in Setup()
210 convBwdInputDims.strides, convBwdInputDims.dilations, in Setup()
282 key_creator.AddAsKey(convBwdInputDims.dilations); in CreateKey()
371 memory::dims padding_left, padding_right, dilations, strides; in Compute() local
379 &strides, &dilations, &fwd_output_dims_tf_order, &fwd_output_dims, in Compute()
418 for (int i = 0; i < dilations.size(); ++i) --dilations[i]; in Compute()
421 native_format, dilations, padding_left, padding_right); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 63 memory::dims dilations; member
71 memory::dims dilations, memory::dims padding_left, in MklConvBwdFilterParams()
80 dilations(dilations), in MklConvBwdFilterParams()
234 convBwdFilterDims.strides, convBwdFilterDims.dilations, in Setup()
244 convBwdFilterDims.dilations, convBwdFilterDims.padding_left, in Setup()
250 convBwdFilterDims.strides, convBwdFilterDims.dilations, in Setup()
337 key_creator.AddAsKey(convBwdFilterDims.dilations); in CreateKey()
419 memory::dims padding_left, padding_right, dilations, strides; in Compute() local
427 &strides, &dilations, &fwd_dst_dims_tf_order, &fwd_dst_dims, in Compute()
465 for (int i = 0; i < dilations.size(); ++i) --dilations[i]; in Compute()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/EDSC/ |
| D | Builders.h | 182 ArrayRef<int> dilations = {});
187 ArrayRef<int> dilations = {}) {
190 dilations);
218 ArrayRef<int> dilations = {});
224 ArrayRef<int> dilations = {}) {
227 depth_multiplier, strides, dilations);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | fused_kernel_matcher.mlir | 9 …// CHECK: %[[VAL_0:.*]] = "tf._FusedConv2D"(%arg2, %arg1, %arg0) {data_format = "NHWC", dilations …
12 …%0 = "tf.Conv2D"(%arg2, %arg1) {data_format = "NHWC", dilations = [1, 1, 1, 1], explicit_paddings …
20 …// CHECK: %[[VAL_0:.*]] = "tf._FusedConv2D"(%arg2, %arg1, %arg0) {data_format = "NHWC", dilations …
23 …%0 = "tf.Conv2D"(%arg2, %arg1) {data_format = "NHWC", dilations = [1, 1, 1, 1], explicit_paddings …
32 …// CHECK: %[[VAL_0:.*]] = "tf._FusedConv2D"(%arg2, %arg1, %arg0) {data_format = "NHWC", dilations …
35 …%0 = "tf.Conv2D"(%arg2, %arg1) {data_format = "NHWC", dilations = [1, 1, 1, 1], explicit_paddings …
44 …// CHECK: %[[VAL_0:.*]] = "tf._FusedConv2D"(%arg2, %arg1, %arg0) {data_format = "NHWC", dilations …
47 …%0 = "tf.Conv2D"(%arg2, %arg1) {data_format = "NHWC", dilations = [1, 1, 1, 1], explicit_paddings …
57 …%0 = "tf.Conv2D"(%arg2, %arg1) {data_format = "NHWC", dilations = [1, 1, 1, 1], explicit_paddings …
67 …AL:.*]] = "tf._FusedConv2D"(%arg2, %arg1, %arg0) {data_format = "NHWC", dilations = [1, 1, 1, 1], …
[all …]
|
| D | layout_optimization_layout_assignment_to_nchw.mlir | 4 // convolution parameters (stride, dilations, etc...). This test only verifies
15 // CHECK-SAME: dilations = [1, 4, 2, 3]
28 dilations = [1, 2, 3, 4],
46 // CHECK-SAME: dilations = [1, 4, 2, 3]
60 dilations = [1, 2, 3, 4],
87 // CHECK-SAME: dilations = [1, 4, 2, 3]
99 dilations = [1, 2, 3, 4],
126 // CHECK-SAME: dilations = [1, 4, 2, 3]
140 dilations = [1, 2, 3, 4],
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 113 const int attrs_dilations_size = attrs.dilations.size(); in CheckConvAttrs()
118 if (attrs.dilations[batch_dim] != 1 || attrs.dilations[feature_dim] != 1) { in CheckConvAttrs()
125 if (attrs.dilations[input_dim] < 1) { in CheckConvAttrs()
128 attrs.dilations[input_dim]); in CheckConvAttrs()
139 absl::Span<const int32> dilations, const std::vector<int32>& strides, in ConvBackpropComputeDimensionsV2XlaShapes() argument
150 out_backprop_tensor_shape, dilations, strides, padding, explicit_paddings, in ConvBackpropComputeDimensionsV2XlaShapes()
166 TF_RETURN_IF_ERROR(ctx->GetAttr("dilations", &attrs.dilations)); in Create()
263 rhs_dilation[i] = attrs.dilations.at(dim); in MakeXlaForwardConvOp()
320 out_backprop_shape, attrs.dilations, attrs.strides, attrs.padding, in MakeXlaBackpropInputConvOp()
364 rhs_dilation[i] = attrs.dilations[dim]; in MakeXlaBackpropInputConvOp()
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 69 …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/core/kernels/ |
| D | conv_ops_3d.cc | 60 const std::array<int64, 3>& dilations, in launch()
68 dilations[0] == 1 && dilations[1] == 1 && dilations[2] == 1, in launch()
157 std::array<int64, 3> dilations = { in Compute() local
167 context, Get3dOutputSizeV2(input_size, filter_size, dilations, strides, in Compute()
178 dilations, strides, padding_, data_format_, in Compute()
214 const std::array<int64, 3>& dilations, in launch()
252 filter_cols == 1 && dilations[0] == 1 && dilations[1] == 1 && in launch()
253 dilations[2] == 1 && strides[0] == 1 && strides[1] == 1 && in launch()
419 conv_desc.set_dilation_rate(DimIndex::X, dilations[2]) in launch()
420 .set_dilation_rate(DimIndex::Y, dilations[1]) in launch()
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 69 …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/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 63 dilations=[1, dilation_w, dilation_h, 1],
96 dilations = [1, op.get_attr('dilation_w'), op.get_attr('dilation_h'), 1]
107 dilations=dilations,
115 dilations=dilations,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | depthwise_conv.cc | 57 {"dilation_w", attr.dilations.w}, in GenerateCode()
58 {"dilation_h", attr.dilations.h}, in GenerateCode()
69 offsets.emplace_back(w * attr.dilations.w - attr.padding.prepended.w, in GenerateCode()
70 h * attr.dilations.h - attr.padding.prepended.h); in GenerateCode()
|
| D | conv.cc | 58 {"dilation_w", attr.dilations.w}, in GenerateCode()
59 {"dilation_h", attr.dilations.h}, in GenerateCode()
69 offsets.emplace_back(w * attr.dilations.w - attr.padding.prepended.w, in GenerateCode()
70 h * attr.dilations.h - attr.padding.prepended.h); in GenerateCode()
176 if (attr.dilations.h != 1 || attr.dilations.w != 1) { in GenerateCode()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/EDSC/ |
| D | Builders.cpp | 212 ArrayRef<int> dilations) { in linalg_generic_conv_nhwc() argument
215 assert((dilations.empty() || dilations.size() == 2) && "only 2-D conv atm"); in linalg_generic_conv_nhwc()
222 auto d = dilations; in linalg_generic_conv_nhwc()
248 ArrayRef<int> dilations) { in linalg_generic_dilated_conv_nhwc() argument
251 assert((dilations.empty() || dilations.size() == 2) && "only 2-D conv atm"); in linalg_generic_dilated_conv_nhwc()
258 auto d = dilations; in linalg_generic_dilated_conv_nhwc()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | depthwise_conv.cc | 245 op.args_.AddInt("dilation_x", attr.dilations.w); in CreateDepthwiseConvolution2D()
249 op.args_.AddInt("dilation_y", attr.dilations.h); in CreateDepthwiseConvolution2D()
278 op.args_.AddInt("dilation_x", attr.dilations.w); in CreateDepthwiseConvolution2DDynamicWeights()
281 op.args_.AddInt("dilation_y", attr.dilations.h); in CreateDepthwiseConvolution2DDynamicWeights()
309 op.args_.AddInt("dilation_x", attr.dilations.w); in CreateDepthwiseConvolution3D()
313 op.args_.AddInt("dilation_y", attr.dilations.h); in CreateDepthwiseConvolution3D()
317 op.args_.AddInt("dilation_z", attr.dilations.d); in CreateDepthwiseConvolution3D()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | conv2d_test.py | 47 dilations = [1] + list(dilation_rate) + [1]
51 dilations = [1, 1] + list(dilation_rate)
58 dilations=dilations,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | prepare-tf-with-allowing-bf16-and-f16-type-legalization.mlir | 7 …%0 = "tf.Conv2D"(%arg0, %arg1) {T = "tfdtype$DT_FLOAT", data_format = "NHWC", dilations = [1, 2, 3…
21 …v2dNative"(%arg0, %arg1) {T = "tfdtype$DT_FLOAT", data_format = "NHWC", dilations = [1, 2, 3, 1], …
28 …%0 = "tf.Conv2D"(%arg0, %arg1) {T = "tfdtype$DT_FLOAT", data_format = "NHWC", dilations = [1, 2, 3…
|