| /external/tensorflow/tensorflow/core/util/ |
| D | padding.cc | 37 const std::vector<int64>& explicit_paddings, in CheckValidPadding() argument
40 const int num_paddings = explicit_paddings.size(); in CheckValidPadding()
44 " values, but got: ", explicit_paddings.size()); in CheckValidPadding()
46 for (int64 padding_value : explicit_paddings) { in CheckValidPadding()
54 if (explicit_paddings[2 * batch_index] != 0 || in CheckValidPadding()
55 explicit_paddings[2 * batch_index + 1] != 0 || in CheckValidPadding()
56 explicit_paddings[2 * depth_index] != 0 || in CheckValidPadding()
57 explicit_paddings[2 * depth_index + 1] != 0) { in CheckValidPadding()
62 } else if (!explicit_paddings.empty()) { in CheckValidPadding()
|
| D | tensor_format.h | 499 const std::vector<int64>& explicit_paddings, TensorFormat tensor_format, in GetExplicitPaddingForDim() argument
502 GetTensorDimIndex(tensor_format, dimension, explicit_paddings.size() / 2); in GetExplicitPaddingForDim()
503 *padding_before = explicit_paddings[2 * index]; in GetExplicitPaddingForDim()
504 *padding_after = explicit_paddings[2 * index + 1]; in GetExplicitPaddingForDim()
|
| D | padding.h | 51 const std::vector<int64>& explicit_paddings,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 614 const std::vector<int>& explicit_paddings, in RunConv2DWithBias() argument
624 ops::Conv2D::Attrs().ExplicitPaddings(explicit_paddings)); in RunConv2DWithBias()
636 const std::string& padding, const std::vector<int>& explicit_paddings, in RunConv2DWithBiasAndActivation() argument
645 ops::Conv2D::Attrs().ExplicitPaddings(explicit_paddings)); in RunConv2DWithBiasAndActivation()
670 const std::string& padding, const std::vector<int>& explicit_paddings, in RunConv2DWithBatchNorm() argument
679 ops::Conv2D::Attrs().ExplicitPaddings(explicit_paddings)); in RunConv2DWithBatchNorm()
700 const std::vector<int>& explicit_paddings, Tensor* output, in RunConv2DWithBatchNormAndActivation() argument
709 ops::Conv2D::Attrs().ExplicitPaddings(explicit_paddings)); in RunConv2DWithBatchNormAndActivation()
743 const std::vector<int>& explicit_paddings, in RunFusedConv2DOp() argument
772 .Attr("explicit_paddings", explicit_paddings) in RunFusedConv2DOp()
[all …]
|
| D | conv_grad_ops.h | 178 const std::vector<int64>& explicit_paddings,
188 const std::vector<int64>& explicit_paddings,
199 const std::vector<int64>& explicit_paddings, Tensor* output,
209 const std::vector<int64>& explicit_paddings,
|
| D | conv_ops.h | 40 const std::vector<int64>& explicit_paddings, Tensor* output,
51 const std::vector<int64>& explicit_paddings, Tensor* output,
77 std::vector<int64> explicit_paddings;
|
| D | conv_ops.cc | 83 const std::vector<int64>& explicit_paddings, Tensor* output, in operator ()()
128 row_dilation, col_dilation, static_cast<int>(explicit_paddings[2]), in operator ()()
129 static_cast<int>(explicit_paddings[3]), in operator ()()
130 static_cast<int>(explicit_paddings[4]), in operator ()()
131 static_cast<int>(explicit_paddings[5])); in operator ()()
149 const std::vector<int64>& explicit_paddings, Tensor* output, in operator ()()
167 for (int64 explicit_padding : explicit_paddings) { in operator ()()
175 explicit_paddings, output, data_format); in operator ()()
186 const std::vector<int64>& explicit_paddings, Tensor* output, in operator ()()
205 for (int64 explicit_padding : explicit_paddings) { in operator ()()
[all …]
|
| D | pooling_ops_common_gpu.h | 46 std::vector<int64> explicit_paddings,
62 std::vector<int64> explicit_paddings,
|
| D | conv_grad_shape_utils.cc | 98 Padding padding, absl::Span<const int64> explicit_paddings, in ConvBackpropComputeDimensionsV2() argument
144 padding_before = explicit_paddings[2 * image_dim]; in ConvBackpropComputeDimensionsV2()
145 padding_after = explicit_paddings[2 * image_dim + 1]; in ConvBackpropComputeDimensionsV2()
|
| D | pooling_ops_common.cc | 120 std::vector<int64> explicit_paddings, in PoolParameters() argument
153 OP_REQUIRES_OK(context, CheckValidPadding(padding, explicit_paddings, in PoolParameters()
155 GetExplicitPaddingForDim(explicit_paddings, data_format, 'H', &pad_top, in PoolParameters()
157 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', &pad_left, in PoolParameters()
217 std::vector<int64> explicit_paddings, in Compute() argument
230 explicit_paddings, data_format, tensor_in.shape()}; in Compute()
444 Padding padding, std::vector<int64> explicit_paddings, in Compute() argument
461 explicit_paddings, data_format, tensor_in_shape}; in Compute()
|
| D | conv_grad_input_ops.cc | 55 const std::vector<int64>& explicit_paddings, in operator ()()
60 explicit_paddings, in_backprop, data_format); in operator ()()
70 const std::vector<int64>& explicit_paddings, Tensor* in_backprop, in operator ()() argument
92 explicit_paddings, data_format, &dims)); in operator ()()
97 GetExplicitPaddingForDim(explicit_paddings, data_format, 'H', &padding_top, in operator ()()
99 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', &padding_left, in operator ()()
|
| D | conv_grad_input_ops.h | 115 const std::vector<int64>& explicit_paddings, in operator()
135 padding, explicit_paddings, data_format, &dims)); in operator()
140 GetExplicitPaddingForDim(explicit_paddings, data_format, 'H', in operator()
142 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', in operator()
210 const std::vector<int64>& explicit_paddings,
215 explicit_paddings, in_backprop, data_format);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | fused_kernel_matcher.mlir | 9 …t = "NHWC", dilations = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_…
12 …%0 = "tf.Conv2D"(%arg2, %arg1) {data_format = "NHWC", dilations = [1, 1, 1, 1], explicit_paddings …
20 …t = "NHWC", dilations = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_…
23 …%0 = "tf.Conv2D"(%arg2, %arg1) {data_format = "NHWC", dilations = [1, 1, 1, 1], explicit_paddings …
32 …t = "NHWC", dilations = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_…
35 …%0 = "tf.Conv2D"(%arg2, %arg1) {data_format = "NHWC", dilations = [1, 1, 1, 1], explicit_paddings …
44 …t = "NHWC", dilations = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_…
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 …t = "NHWC", dilations = [1, 1, 1, 1], epsilon = 0.000000e+00 : f32, explicit_paddings = [], fused_…
[all …]
|
| D | layout_optimization_layout_assignment_to_nchw.mlir | 16 // CHECK-SAME: explicit_paddings = [1, 2, 7, 8, 3, 4, 5, 6]
29 explicit_paddings = [1, 2, 3, 4, 5, 6, 7, 8],
47 // CHECK-SAME: explicit_paddings = [1, 2, 7, 8, 3, 4, 5, 6]
61 explicit_paddings = [1, 2, 3, 4, 5, 6, 7, 8],
88 // CHECK-SAME: explicit_paddings = [1, 2, 7, 8, 3, 4, 5, 6]
100 explicit_paddings = [1, 2, 3, 4, 5, 6, 7, 8],
127 // CHECK-SAME: explicit_paddings = [1, 2, 7, 8, 3, 4, 5, 6]
141 explicit_paddings = [1, 2, 3, 4, 5, 6, 7, 8],
|
| D | layout_optimization_layout_assignment_to_nhwc.mlir | 16 // CHECK-SAME: explicit_paddings = [1, 2, 5, 6, 7, 8, 3, 4]
29 explicit_paddings = [1, 2, 3, 4, 5, 6, 7, 8],
90 …// CHECK: %[[R1:.*]] = "tf.MaxPool"(%[[R0]]) {data_format = "NHWC", explicit_paddings = [], ksize …
|
| D | layout_optimization_to_nhwc.mlir | 50 explicit_paddings = [],
93 explicit_paddings = [],
120 explicit_paddings = [],
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Conv2DBackpropInput.pbtxt | 46 name: "explicit_paddings"
50 `explicit_paddings[2 * i]` and `explicit_paddings[2 * i + 1]`, respectively. If
51 `padding` is not `"EXPLICIT"`, `explicit_paddings` must be empty.
|
| D | api_def_Conv2DBackpropFilter.pbtxt | 47 name: "explicit_paddings"
51 `explicit_paddings[2 * i]` and `explicit_paddings[2 * i + 1]`, respectively. If
52 `padding` is not `"EXPLICIT"`, `explicit_paddings` must be empty.
|
| D | api_def_Conv2D.pbtxt | 39 name: "explicit_paddings"
43 `explicit_paddings[2 * i]` and `explicit_paddings[2 * i + 1]`, respectively. If
44 `padding` is not `"EXPLICIT"`, `explicit_paddings` must be empty.
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 52 explicit_paddings=op.get_attr("explicit_paddings"),
61 explicit_paddings=op.get_attr("explicit_paddings"),
79 explicit_paddings=op.get_attr("explicit_paddings"),
88 explicit_paddings=op.get_attr("explicit_paddings"),
114 explicit_paddings=op.get_attr("explicit_paddings"),
122 explicit_paddings=op.get_attr("explicit_paddings"),
137 explicit_paddings=op.get_attr("explicit_paddings"),
145 explicit_paddings=op.get_attr("explicit_paddings"),
575 explicit_paddings = op.get_attr("explicit_paddings")
594 explicit_paddings=explicit_paddings,
[all …]
|
| D | nn_ops.py | 653 padding, explicit_paddings = convert_padding(padding)
676 base_paddings = (np.array(explicit_paddings)
1770 explicit_paddings = []
1786 explicit_paddings.extend(dim_paddings)
1791 return padding, explicit_paddings
2252 padding, explicit_paddings = convert_padding(padding)
2275 explicit_paddings=explicit_paddings,
2287 explicit_paddings=explicit_paddings,
2349 padding, explicit_paddings = convert_padding(padding)
2352 explicit_paddings, data_format, dilations, name)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 141 absl::Span<const int64> explicit_paddings) { in ConvBackpropComputeDimensionsV2XlaShapes() argument
150 out_backprop_tensor_shape, dilations, strides, padding, explicit_paddings, in ConvBackpropComputeDimensionsV2XlaShapes()
171 ctx->GetAttr("explicit_paddings", &attrs.explicit_paddings)); in Create()
180 TF_RETURN_IF_ERROR(CheckValidPadding(attrs.padding, attrs.explicit_paddings, in Create()
266 padding[i] = {attrs.explicit_paddings.at(dim * 2), in MakeXlaForwardConvOp()
267 attrs.explicit_paddings.at(dim * 2 + 1)}; in MakeXlaForwardConvOp()
321 attrs.data_format, &dims, attrs.explicit_paddings)); in MakeXlaBackpropInputConvOp()
425 attrs.padding, attrs.data_format, &dims, attrs.explicit_paddings)); in MakeXlaBackpropFilterConvOp()
513 ? attrs.explicit_paddings[2 * dim] in MakeXlaBackpropFilterConvOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/tests/ |
| D | fuse-bias-tf.mlir | 12 …arg1) {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], paddin…
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Conv2D.pbtxt | 254 name: "explicit_paddings"
337 name: "explicit_paddings"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Conv2D.pbtxt | 254 name: "explicit_paddings"
337 name: "explicit_paddings"
|