Home
last modified time | relevance | path

Searched defs:explicit_paddings (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dreplace_cast_hacks_with_tf_xla_ops.cc99 StringAttr conv_padding, ArrayAttr explicit_paddings, in CreateXLAConvOp()
166 ArrayAttr explicit_paddings) { in CreateXLAConvOpFromTFConv2DOp()
186 StringAttr conv_padding, ArrayAttr explicit_paddings) { in CreateXLAConvOpFromTFDepthwiseConv2DOp()
/external/tensorflow/tensorflow/core/util/
Dpadding.cc37 const std::vector<int64_t>& explicit_paddings, in CheckValidPadding()
Dtensor_format.h499 const std::vector<int64_t>& explicit_paddings, TensorFormat tensor_format, in GetExplicitPaddingForDim()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.h54 std::vector<int64_t> explicit_paddings; member
Dconv_op_helpers.cc141 absl::Span<const int64_t> explicit_paddings) { in ConvBackpropComputeDimensionsV2XlaShapes()
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_common.cc121 std::vector<int64_t> explicit_paddings, in PoolParameters()
220 std::vector<int64_t> explicit_paddings, in Compute()
443 Padding padding, std::vector<int64_t> explicit_paddings, in Compute()
Dconv_ops_test.cc557 const std::vector<int>& explicit_paddings, in RunConv2DWithBias()
579 const std::string& padding, const std::vector<int>& explicit_paddings, in RunConv2DWithBiasAndActivation()
613 const std::string& padding, const std::vector<int>& explicit_paddings, in RunConv2DWithBatchNorm()
643 const std::vector<int>& explicit_paddings, Tensor* output, in RunConv2DWithBatchNormAndActivation()
686 const std::vector<int>& explicit_paddings, in RunFusedConv2DOp()
816 const std::vector<int>& explicit_paddings = {}, in VerifyConv2DWithBias()
846 const std::vector<int>& explicit_paddings = {}, int depth = kDepth, in VerifyConv2DWithBiasAndActivation()
878 const std::vector<int>& explicit_paddings = {}, in VerifyConv2DWithBatchNorm()
913 const std::vector<int>& explicit_paddings = {}, int depth = kDepth, in VerifyConv2DWithBatchNormAndActivation()
Dconv_ops_fused_impl.h100 Padding padding, const std::vector<int64_t>& explicit_paddings) in LaunchFusedConv2DWithOutputKernel()
Dconv_grad_shape_utils.cc99 Padding padding, absl::Span<const int64_t> explicit_paddings, in ConvBackpropComputeDimensionsV2()
Dconv_grad_input_ops.cc77 const std::vector<int64_t>& explicit_paddings, Tensor* in_backprop, in operator ()()
Dconv_grad_filter_ops.cc673 const std::vector<int64_t>& explicit_paddings, Tensor* filter_backprop, in operator ()()
Dconv_ops.cc805 const std::vector<int64_t>& explicit_paddings, Tensor* output, in operator ()()
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
Dtf_to_xla_attribute_utils.cc140 StringAttr conv_padding, ArrayAttr explicit_paddings, Value &padding) { in CalculatePaddingAndPadIfNeeded()
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad.cc523 std::vector<int> dilations, strides, explicit_paddings; in Conv2DBackpropInputGrad() local
569 std::vector<int> dilations, strides, explicit_paddings; in DepthwiseConv2dNativeGrad() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns_test.cc631 const std::vector<int32>& explicit_paddings = {}) { in TEST()
824 const std::vector<int32>& explicit_paddings = {}) { in TEST()
Dcommon_shape_fns.cc698 std::vector<int64_t> explicit_paddings; in Conv2DShapeImpl() local
1053 std::vector<int64_t> explicit_paddings; in DepthwiseConv2DNativeShapeImpl() local
1639 std::vector<int64_t> explicit_paddings; in MaxPoolShapeImpl() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1140 ArrayRef<Attribute> explicit_paddings; in matchAndRewriteDynamicConv() local
1303 ArrayRef<Attribute> explicit_paddings; in matchAndRewrite() local
4928 std::vector<int64_t> explicit_paddings; in matchAndRewrite() local
5127 std::vector<int64_t> explicit_paddings; in matchAndRewrite() local
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc2619 std::vector<int32> explicit_paddings; in CopyAttrsConv() local