Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc31 int64 padding_bottom, int64 padding_left, int64 padding_right, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
38 padding_top, padding_bottom, padding_left, padding_right, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
49 int64 padding_left, int64 padding_right, int64 lhs_row_dilation, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
55 padding_top, padding_bottom, padding_left, padding_right, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc33 int64 padding_left, int64 padding_right, int64 lhs_row_dilation, in __xla_cpu_runtime_EigenConvF32() argument
41 col_stride, padding_top, padding_bottom, padding_left, padding_right, in __xla_cpu_runtime_EigenConvF32()
51 int64 padding_bottom, int64 padding_left, int64 padding_right, in __xla_cpu_runtime_EigenConvF16() argument
60 col_stride, padding_top, padding_bottom, padding_left, padding_right, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc58 int64 padding_left, int64 padding_right, in MKLConvImpl() argument
80 memory::dims conv1_padding_r = {ToInt(padding_bottom), ToInt(padding_right)}; in MKLConvImpl()
158 int64 padding_left, int64 padding_right, int64 lhs_row_dilation, in __xla_cpu_runtime_MKLConvF32() argument
168 padding_top, padding_bottom, padding_left, padding_right, in __xla_cpu_runtime_MKLConvF32()
175 padding_right, lhs_row_dilation, lhs_col_dilation, in __xla_cpu_runtime_MKLConvF32()
Druntime_single_threaded_conv2d.h34 tensorflow::int64 padding_left, tensorflow::int64 padding_right,
48 tensorflow::int64 padding_right, tensorflow::int64 lhs_row_dilation,
Druntime_conv2d.h34 tensorflow::int64 padding_right, tensorflow::int64 lhs_row_dilation,
48 tensorflow::int64 padding_left, tensorflow::int64 padding_right,
Druntime_conv2d_impl.h35 int64 padding_right, int64 lhs_row_dilation, in EigenConvImpl() argument
78 padding_left, padding_right, padding_top, in EigenConvImpl()
Druntime_conv2d_mkl.h34 tensorflow::int64 padding_right, tensorflow::int64 lhs_row_dilation,
/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_grad_input_ops.cc61 memory::dims padding_right; member
67 memory::dims padding_right, padding_kind padding) in MklConvBwdInputParams()
74 padding_right(padding_right), in MklConvBwdInputParams()
191 convBwdInputDims.padding_right, convBwdInputDims.padding)); in Setup()
197 convBwdInputDims.padding_right, convBwdInputDims.padding)); in Setup()
282 key_creator.AddAsKey(convBwdInputDims.padding_right); in CreateKey()
362 memory::dims padding_left, padding_right, dilations, strides; in Compute() local
371 &padding_left, &padding_right, false, is_depthwise); in Compute()
404 padding_left, padding_right, in Compute()
Dmkl_conv_grad_filter_ops.cc59 memory::dims padding_right; member
66 memory::dims padding_right, padding_kind padding) in MklConvBwdFilterParams()
74 padding_right(padding_right), in MklConvBwdFilterParams()
232 convBwdFilterDims.padding_left, convBwdFilterDims.padding_right, in Setup()
239 convBwdFilterDims.padding_right, convBwdFilterDims.padding)); in Setup()
247 convBwdFilterDims.padding_left, convBwdFilterDims.padding_right, in Setup()
343 key_creator.AddAsKey(convBwdFilterDims.padding_right); in CreateKey()
425 memory::dims padding_left, padding_right, dilations, strides, in Compute() local
435 &padding_left, &padding_right, false, is_depthwise); in Compute()
469 dilations, padding_left, padding_right, in Compute()
Deigen_backward_spatial_convolutions.h160 const TensorIndex padding_right = inputCols - (outputCols - 1) * col_stride - variable
166 eigen_assert(padding_right >= 0);
282 padding_bottom, padding_left, padding_right, OutScalar(0))
290 padding_right, OutScalar(0))
416 const TensorIndex padding_right = padCols - padding_left; variable
486 padding_bottom, padding_left, padding_right, OutScalar(0))
494 padding_right, OutScalar(0))
Dmkl_pooling_ops_common.h51 memory::dims padding_right; member
57 memory::dims padding_left, memory::dims padding_right, in MklPoolingParams()
64 padding_right(padding_right), in MklPoolingParams()
193 key_creator.AddAsKey(fwdParams.padding_right); in CreateKey()
345 key_creator.AddAsKey(bwdParams.padding_right); in CreateKey()
523 memory::dims* padding_left, memory::dims* padding_right, in PoolParamsToDims() argument
533 *padding_right = memory::dims({static_cast<int>(pool_params->pad_bottom), in PoolParamsToDims()
547 *padding_right = memory::dims({static_cast<int>(pool_params->pad_P2), in PoolParamsToDims()
Dmkl_avgpooling_op.cc477 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
480 &padding_left, &padding_right, is_pool2d); in Compute()
505 strides, padding_left, padding_right, in Compute()
599 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
601 &padding_left, &padding_right, is_pool2d); in Compute()
625 strides, padding_left, padding_right, in Compute()
Dconv_grad_input_ops.cc709 int64 padding_left = -1, padding_right = -1; in operator ()() local
714 &padding_right); in operator ()()
727 &padding_right)); in operator ()()
811 const int64 common_padding_cols = std::min(padding_left, padding_right); in operator ()()
813 if (padding_top != padding_bottom || padding_left != padding_right) { in operator ()()
818 const int64 padding_cols_diff = std::abs(padding_right - padding_left); in operator ()()
1010 if (padding_top != padding_bottom || padding_left != padding_right) { in operator ()()
1026 const int64 input_pad_right = padding_right - common_padding_cols; in operator ()()
1084 const std::array<int, 2>& padding_right, \
Dconv_ops.cc672 int64 padding_left = -1, padding_right = -1; in operator ()() local
677 &padding_right); in operator ()()
689 &padding_left, &padding_right); in operator ()()
694 const int64 common_padding_cols = std::min(padding_left, padding_right); in operator ()()
695 if (padding_top != padding_bottom || padding_left != padding_right) { in operator ()()
707 const int64 padding_cols_diff = std::abs(padding_right - padding_left); in operator ()()
719 const int64 input_pad_right = padding_right - common_padding_cols; in operator ()()
948 const std::array<int, 2>& padding_right, \
Dconv_grad_filter_ops.cc570 int64 padding_left = -1, padding_right = -1; in operator ()() local
575 &padding_right); in operator ()()
588 &padding_right)); in operator ()()
680 const int64 common_padding_cols = std::min(padding_left, padding_right); in operator ()()
682 if (padding_top != padding_bottom || padding_left != padding_right) { in operator ()()
687 const int64 padding_cols_diff = std::abs(padding_right - padding_left); in operator ()()
695 const int64 input_pad_right = padding_right - common_padding_cols; in operator ()()
941 const std::array<int, 2>& padding_right, \
Dmkl_maxpooling_op.cc568 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
570 &padding_left, &padding_right, is_pool2d); in Compute()
582 strides, padding_left, padding_right, in Compute()
710 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
712 &padding_left, &padding_right, is_pool2d); in Compute()
735 strides, padding_left, padding_right, algorithm::pooling_max, in Compute()
Dmkl_conv_ops.cc74 memory::dims padding_right; member
85 memory::dims padding_left, memory::dims padding_right) in MklConvFwdParams()
93 padding_right(padding_right) {} in MklConvFwdParams()
235 convFwdDims.padding_right, padding_kind::zero)); in Setup()
241 convFwdDims.padding_right, padding_kind::zero)); in Setup()
372 key_creator.AddAsKey(convFwdDims.padding_right); in CreateKey()
940 memory::dims src_dims, filter_dims, padding_left, padding_right, in Compute() local
955 PadWithConvFusion(context, padding_left, padding_right, in Compute()
967 &padding_right, (fuse_pad_ || quantized_pad_enabled), is_depthwise); in Compute()
1074 dst_dims_mkl_order, strides, dilations, padding_left, padding_right); in Compute()
[all …]
Deigen_backward_cuboid_convolutions.h160 const TensorIndex padding_right = inputCols - (outputCols - 1) * col_stride - variable
168 eigen_assert(padding_right >= 0);
285 padding_left, padding_right)
293 padding_bottom, padding_left, padding_right)
Dmkl_pooling_ops_common.cc66 fwdParams.padding_left, fwdParams.padding_right, padding_kind::zero)); in Setup()
153 bwdParams.padding_right, padding_kind::zero)); in Setup()
160 bwdParams.padding_left, bwdParams.padding_right, padding_kind::zero)); in Setup()
Dconv_2d.h251 const std::array<int, NDIMS - 2>& padding_right,
258 padding_left[i], padding_right[i]};
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorImagePatch.h85 … DenseIndex padding_left, DenseIndex padding_right,
92 m_padding_left(padding_left), m_padding_right(padding_right),
120 DenseIndex padding_right() const { return m_padding_right; }
222 …m_outputCols = numext::ceil((m_input_cols_eff + op.padding_left() + op.padding_right() - m_patch_c…
DTensorVolumePatch.h81 … DenseIndex padding_left, DenseIndex padding_right,
88 m_padding_left(padding_left), m_padding_right(padding_right),
128 DenseIndex padding_right() const { return m_padding_right; }
232 …m_outputCols = numext::ceil((m_input_cols_eff + op.padding_left() + op.padding_right() - m_patch_c…
DTensorBase.h703 const Index padding_left,const Index padding_right, in extract_image_patches() argument
707 … padding_top, padding_bottom, padding_left, padding_right, padding_value); in extract_image_patches()
726 …const Index padding_left, const Index padding_right, const Scalar padding_value = Scalar(0)) const…
727 …g_top_z, padding_bottom_z, padding_top, padding_bottom, padding_left, padding_right, padding_value…
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h433 const int padding_right = function_params.padding_right;
440 padding_right > 0 &&
627 const int padding_right = function_params->padding_right;
634 padding_right > 0 &&
900 const int padding_right = function_params->padding_right;
911 padding_right > 0 &&
1203 TFLITE_DCHECK_EQ(function_params->padding_right, 0);
1406 const int padding_right = function_params->padding_right;
1425 padding_right > 0 &&
1663 const int padding_right = function_params->padding_right;
[all …]
Ddepthwiseconv_uint8_3x3_filter.h397 int32 padding_right;
3920 TFLITE_DCHECK_EQ(function_params->padding_right, 0);
4119 const int padding_right = function_params->padding_right;
4138 padding_right > 0 &&
4372 const int padding_right = function_params->padding_right;
4383 padding_right > 0 &&
4725 TFLITE_DCHECK_EQ(function_params->padding_right, 0);
7144 const int padding_right =
7152 function_params.padding_right = padding_right;
7156 TFLITE_DCHECK_LE(padding_left, padding_right);
[all …]

12