Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dpadding.cc41 const std::vector<int64>& explicit_paddings, in CheckValidPadding() argument
44 if (explicit_paddings.size() != 2 * num_dims) { in CheckValidPadding()
47 " values, but got: ", explicit_paddings.size()); in CheckValidPadding()
49 for (int64 padding_value : explicit_paddings) { in CheckValidPadding()
57 if (explicit_paddings[2 * batch_index] != 0 || in CheckValidPadding()
58 explicit_paddings[2 * batch_index + 1] != 0 || in CheckValidPadding()
59 explicit_paddings[2 * depth_index] != 0 || in CheckValidPadding()
60 explicit_paddings[2 * depth_index + 1] != 0) { in CheckValidPadding()
65 } else if (!explicit_paddings.empty()) { in CheckValidPadding()
Dtensor_format.h486 const std::vector<int64>& explicit_paddings, TensorFormat tensor_format, in GetExplicitPaddingForDim() argument
489 GetTensorDimIndex(tensor_format, dimension, explicit_paddings.size() / 2); in GetExplicitPaddingForDim()
490 *padding_before = explicit_paddings[2 * index]; in GetExplicitPaddingForDim()
491 *padding_after = explicit_paddings[2 * index + 1]; in GetExplicitPaddingForDim()
Dpadding.h50 const std::vector<int64>& explicit_paddings,
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_ops.h180 const std::vector<int64>& explicit_paddings,
190 const std::vector<int64>& explicit_paddings,
201 const std::vector<int64>& explicit_paddings, Tensor* output,
211 const std::vector<int64>& explicit_paddings,
273 Padding padding, absl::Span<const int64> explicit_paddings,
Dconv_ops.h40 const std::vector<int64>& explicit_paddings, Tensor* output,
51 const std::vector<int64>& explicit_paddings, Tensor* output,
77 std::vector<int64> explicit_paddings;
Dconv_grad_ops.cc102 Padding padding, absl::Span<const int64> explicit_paddings, in ConvBackpropComputeDimensionsV2() argument
148 padding_before = explicit_paddings[2 * image_dim]; in ConvBackpropComputeDimensionsV2()
149 padding_after = explicit_paddings[2 * image_dim + 1]; in ConvBackpropComputeDimensionsV2()
Dconv_ops.cc128 const std::vector<int64>& explicit_paddings, Tensor* output, in operator ()()
287 context->GetAttr("explicit_paddings", &params->explicit_paddings)); in InitConv2DParameters()
329 params->explicit_paddings, in InitConv2DParameters()
402 GetExplicitPaddingForDim(params.explicit_paddings, params.data_format, 'H', in ComputeConv2DDimension()
404 GetExplicitPaddingForDim(params.explicit_paddings, params.data_format, 'W', in ComputeConv2DDimension()
520 params_.explicit_paddings, output, params_.data_format); in Compute()
582 const std::vector<int64>& explicit_paddings, Tensor* output, in operator ()() argument
674 GetExplicitPaddingForDim(explicit_paddings, data_format, 'H', &padding_top, in operator ()()
676 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', &padding_left, in operator ()()
Dconv_grad_filter_ops.cc107 const std::vector<int64>& explicit_paddings, in operator ()()
215 std::vector<int64> explicit_paddings; in Conv2DCustomBackpropFilterOp() local
217 context->GetAttr("explicit_paddings", &explicit_paddings)); in Conv2DCustomBackpropFilterOp()
218 OP_REQUIRES_OK(context, CheckValidPadding(padding_, explicit_paddings, in Conv2DCustomBackpropFilterOp()
547 const std::vector<int64>& explicit_paddings, Tensor* filter_backprop, in operator ()() argument
567 strides, padding, explicit_paddings, data_format, &dims)); in operator ()()
572 GetExplicitPaddingForDim(explicit_paddings, data_format, 'H', &padding_top, in operator ()()
574 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', &padding_left, in operator ()()
Dconv_grad_input_ops.cc113 const std::vector<int64>& explicit_paddings, in operator ()()
306 std::vector<int64> explicit_paddings; in Conv2DCustomBackpropInputOp() local
308 context->GetAttr("explicit_paddings", &explicit_paddings)); in Conv2DCustomBackpropInputOp()
309 OP_REQUIRES_OK(context, CheckValidPadding(padding_, explicit_paddings, in Conv2DCustomBackpropInputOp()
684 const std::vector<int64>& explicit_paddings, Tensor* in_backprop, in operator ()() argument
706 explicit_paddings, data_format, &dims)); in operator ()()
711 GetExplicitPaddingForDim(explicit_paddings, data_format, 'H', &padding_top, in operator ()()
713 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', &padding_left, in operator ()()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Conv2DBackpropFilter.pbtxt47 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.
Dapi_def_Conv2DBackpropInput.pbtxt46 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.
Dapi_def_Conv2D.pbtxt39 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/compiler/tf2xla/kernels/
Dconv_op_helpers.cc207 absl::Span<const int64> explicit_paddings) { in ConvBackpropComputeDimensionsV2XlaShapes() argument
216 out_backprop_tensor_shape, dilations, strides, padding, explicit_paddings, in ConvBackpropComputeDimensionsV2XlaShapes()
233 ctx->GetAttr("explicit_paddings", &attrs.explicit_paddings)); in Create()
307 padding[i] = {attrs.explicit_paddings.at(dim * 2), in MakeXlaForwardConvOp()
308 attrs.explicit_paddings.at(dim * 2 + 1)}; in MakeXlaForwardConvOp()
345 attrs.data_format, &dims, attrs.explicit_paddings)); in MakeXlaBackpropInputConvOp()
428 attrs.padding, attrs.data_format, &dims, attrs.explicit_paddings)); in MakeXlaBackpropFilterConvOp()
516 ? attrs.explicit_paddings[2 * dim] in MakeXlaBackpropFilterConvOp()
Dconv_op_helpers.h50 std::vector<int64> explicit_paddings; member
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py1505 explicit_paddings = []
1521 explicit_paddings.extend(dim_paddings)
1526 return padding, explicit_paddings
1936 padding, explicit_paddings = _convert_padding(padding)
1948 explicit_paddings=explicit_paddings,
2008 padding, explicit_paddings = _convert_padding(padding)
2011 explicit_paddings, data_format, dilations, name)
2072 padding, explicit_paddings = _convert_padding(padding)
2075 explicit_paddings, data_format, dilations, name)
Dnn_grad.py570 explicit_paddings = op.get_attr("explicit_paddings")
589 explicit_paddings=explicit_paddings,
599 explicit_paddings=explicit_paddings,
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns_test.cc412 const std::vector<int32>& explicit_paddings = {}) { in TEST() argument
418 .Attr("explicit_paddings", explicit_paddings) in TEST()
600 const std::vector<int32>& explicit_paddings = {}) { in TEST() argument
607 .Attr("explicit_paddings", explicit_paddings) in TEST()
Dcommon_shape_fns.cc504 std::vector<int64> explicit_paddings; in Conv2DShapeImpl() local
506 Status s = c->GetAttr("explicit_paddings", &explicit_paddings); in Conv2DShapeImpl()
512 TF_RETURN_IF_ERROR(CheckValidPadding(padding, explicit_paddings, in Conv2DShapeImpl()
522 GetExplicitPaddingForDim(explicit_paddings, data_format, 'H', in Conv2DShapeImpl()
524 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', in Conv2DShapeImpl()
/external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/
Dlarge_function_graph.pbtxt515 key: "explicit_paddings"
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc154 std::vector<int> explicit_paddings; in SimpleConv2DBackpropInput() local
156 explicit_paddings = {0, in SimpleConv2DBackpropInput()
167 explicit_paddings); in SimpleConv2DBackpropInput()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt661 …ut\', \'filter\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit_paddings\', \'data_fo…
665 …\'out_backprop\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit_paddings\', \'data_fo…
669 …\'out_backprop\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit_paddings\', \'data_fo…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt661 …ut\', \'filter\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit_paddings\', \'data_fo…
665 …\'out_backprop\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit_paddings\', \'data_fo…
669 …\'out_backprop\', \'strides\', \'padding\', \'use_cudnn_on_gpu\', \'explicit_paddings\', \'data_fo…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt6244 name: "explicit_paddings"
6330 name: "explicit_paddings"
6416 name: "explicit_paddings"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt14178 name: "explicit_paddings"
14480 name: "explicit_paddings"
14782 name: "explicit_paddings"
Dops_history.v1.pbtxt14236 name: "explicit_paddings"
14538 name: "explicit_paddings"
14840 name: "explicit_paddings"