/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 …]
|
/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 | 243 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 …]
|
D | nn_grad.py | 49 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 …]
|
D | nn_impl.py | 511 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/ |
D | conv_ops_test.py | 179 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 …]
|
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, 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 …]
|
D | conv2d_backprop_filter_grad_test.py | 84 dilations=[1, 2, 2, 1], 95 dilations=[1, 2, 2, 1],
|
/external/tensorflow/tensorflow/core/kernels/ |
D | mkl_conv_ops.h | 61 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 …]
|
D | conv_ops_3d.cc | 54 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 …]
|
D | mkl_conv_grad_input_ops.cc | 59 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()
|
D | mkl_conv_grad_filter_ops.cc | 57 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 …]
|
D | conv_grad_ops.cc | 60 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()
|
D | quantized_conv_ops.cc | 466 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()
|
D | conv_ops.cc | 282 TF_RETURN_IF_ERROR(context->GetAttr("dilations", ¶ms->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()
|
D | mkl_conv_ops.cc | 72 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/ |
D | conv_op_helpers.cc | 180 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 …]
|
D | conv_op_helpers.h | 47 std::vector<int32> dilations; member
|
/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/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/core/graph/ |
D | mkl_layout_pass.cc | 1532 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/ |
D | conv2d_test.py | 49 dilations = [1] + list(dilation_rate) + [1] 53 dilations = [1, 1] + list(dilation_rate) 60 dilations=dilations,
|
/external/tensorflow/tensorflow/lite/toco/ |
D | import_tensorflow.cc | 662 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/ |
D | common_shape_fns.cc | 121 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/ |
D | dnn.cc | 471 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()
|