Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.h54 std::vector<int64> explicit_paddings; member
Dconv_op_helpers.cc141 absl::Span<const int64> explicit_paddings) { in ConvBackpropComputeDimensionsV2XlaShapes()
/external/tensorflow/tensorflow/core/util/
Dpadding.cc37 const std::vector<int64>& explicit_paddings, in CheckValidPadding()
Dtensor_format.h499 const std::vector<int64>& explicit_paddings, TensorFormat tensor_format, in GetExplicitPaddingForDim()
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_common.cc120 std::vector<int64> explicit_paddings, in PoolParameters()
217 std::vector<int64> explicit_paddings, in Compute()
444 Padding padding, std::vector<int64> explicit_paddings, in Compute()
Dconv_ops_test.cc614 const std::vector<int>& explicit_paddings, in RunConv2DWithBias()
636 const std::string& padding, const std::vector<int>& explicit_paddings, in RunConv2DWithBiasAndActivation()
670 const std::string& padding, const std::vector<int>& explicit_paddings, in RunConv2DWithBatchNorm()
700 const std::vector<int>& explicit_paddings, Tensor* output, in RunConv2DWithBatchNormAndActivation()
743 const std::vector<int>& explicit_paddings, in RunFusedConv2DOp()
873 const std::vector<int>& explicit_paddings = {}, in VerifyConv2DWithBias()
903 const std::vector<int>& explicit_paddings = {}, int depth = kDepth, in VerifyConv2DWithBiasAndActivation()
935 const std::vector<int>& explicit_paddings = {}, in VerifyConv2DWithBatchNorm()
970 const std::vector<int>& explicit_paddings = {}, int depth = kDepth, in VerifyConv2DWithBatchNormAndActivation()
Dconv_ops_fused_impl.h98 const std::vector<int64>& explicit_paddings) in LaunchFusedConv2DWithOutputKernel()
Dconv_grad_shape_utils.cc98 Padding padding, absl::Span<const int64> explicit_paddings, in ConvBackpropComputeDimensionsV2()
Dconv_grad_input_ops.cc70 const std::vector<int64>& explicit_paddings, Tensor* in_backprop, in operator ()()
Dconv_grad_filter_ops.cc654 const std::vector<int64>& explicit_paddings, Tensor* filter_backprop, in operator ()()
Dconv_ops.cc650 const std::vector<int64>& explicit_paddings, Tensor* output, in operator ()()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns_test.cc629 const std::vector<int32>& explicit_paddings = {}) { in TEST()
818 const std::vector<int32>& explicit_paddings = {}) { in TEST()
Dcommon_shape_fns.cc687 std::vector<int64> explicit_paddings; in Conv2DShapeImpl() local
998 std::vector<int64> explicit_paddings; in DepthwiseConv2DNativeShapeImpl() local
1548 std::vector<int64> explicit_paddings; in MaxPoolShapeImpl() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1183 ArrayRef<Attribute> explicit_paddings; in matchAndRewrite() local
4137 std::vector<tensorflow::int64> explicit_paddings; in matchAndRewrite() local
4320 std::vector<tensorflow::int64> explicit_paddings; in matchAndRewrite() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc160 std::vector<int> explicit_paddings; in SimpleConv2DBackpropInput() local
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc2642 std::vector<int32> explicit_paddings; in CopyAttrsConv() local