Home
last modified time | relevance | path

Searched refs:padding_left (Results 1 – 25 of 34) 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
79 memory::dims conv1_padding_l = {ToInt(padding_top), ToInt(padding_left)}; 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()
174 col_stride, padding_top, padding_bottom, padding_left, in __xla_cpu_runtime_MKLConvF32()
Druntime_single_threaded_conv2d.h34 tensorflow::int64 padding_left, tensorflow::int64 padding_right,
47 tensorflow::int64 padding_bottom, tensorflow::int64 padding_left,
Druntime_conv2d.h33 tensorflow::int64 padding_bottom, tensorflow::int64 padding_left,
48 tensorflow::int64 padding_left, tensorflow::int64 padding_right,
Druntime_conv2d_impl.h34 int64 padding_bottom, int64 padding_left, in EigenConvImpl() argument
78 padding_left, padding_right, padding_top, in EigenConvImpl()
Druntime_conv2d_mkl.h33 tensorflow::int64 padding_bottom, tensorflow::int64 padding_left,
/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_grad_input_ops.cc60 memory::dims padding_left; member
66 memory::dims dilations, memory::dims padding_left, in MklConvBwdInputParams()
73 padding_left(padding_left), in MklConvBwdInputParams()
190 convBwdInputDims.dilations, convBwdInputDims.padding_left, in Setup()
196 convBwdInputDims.dilations, convBwdInputDims.padding_left, in Setup()
281 key_creator.AddAsKey(convBwdInputDims.padding_left); 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()
Deigen_backward_spatial_convolutions.h156 const TensorIndex padding_left = kernelColsEff - 1 - forward_pad_left; variable
161 2 - padding_left + kernelColsEff;
164 eigen_assert(padding_left >= 0);
282 padding_bottom, padding_left, padding_right, OutScalar(0))
289 padding_top, padding_bottom, padding_left,
415 const TensorIndex padding_left = padCols / 2; variable
416 const TensorIndex padding_right = padCols - padding_left;
486 padding_bottom, padding_left, padding_right, OutScalar(0))
493 padding_top, padding_bottom, padding_left,
Dmkl_conv_grad_filter_ops.cc58 memory::dims padding_left; member
65 memory::dims dilations, memory::dims padding_left, in MklConvBwdFilterParams()
73 padding_left(padding_left), in MklConvBwdFilterParams()
232 convBwdFilterDims.padding_left, convBwdFilterDims.padding_right, in Setup()
238 convBwdFilterDims.dilations, convBwdFilterDims.padding_left, in Setup()
247 convBwdFilterDims.padding_left, convBwdFilterDims.padding_right, in Setup()
342 key_creator.AddAsKey(convBwdFilterDims.padding_left); 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()
Dmkl_pooling_ops_common.h50 memory::dims padding_left; member
57 memory::dims padding_left, memory::dims padding_right, in MklPoolingParams()
63 padding_left(padding_left), in MklPoolingParams()
192 key_creator.AddAsKey(fwdParams.padding_left); in CreateKey()
344 key_creator.AddAsKey(bwdParams.padding_left); in CreateKey()
523 memory::dims* padding_left, memory::dims* padding_right, in PoolParamsToDims() argument
531 *padding_left = memory::dims({static_cast<int>(pool_params->pad_top), in PoolParamsToDims()
544 *padding_left = memory::dims({static_cast<int>(pool_params->pad_P1), in PoolParamsToDims()
Deigen_backward_cuboid_convolutions.h153 const TensorIndex padding_left = kernelColsEff - 1 - forward_pad_left; variable
161 2 - padding_left + kernelColsEff;
165 eigen_assert(padding_left >= 0);
285 padding_left, padding_right)
293 padding_bottom, padding_left, padding_right)
466 const TensorIndex padding_left = padCols / 2; variable
477 const auto left_pad_cols = kernelCols - 1 - padding_left;
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
713 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', &padding_left, in operator ()()
726 col_dilation, col_stride, padding, &expected_out_cols, &padding_left, 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 ()()
1025 const int64 input_pad_left = padding_left - common_padding_cols; in operator ()()
1083 const std::array<int, 2>& padding_left, \
Dconv_ops.cc672 int64 padding_left = -1, padding_right = -1; in operator ()() local
676 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', &padding_left, 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 ()()
718 const int64 input_pad_left = padding_left - common_padding_cols; in operator ()()
947 const std::array<int, 2>& padding_left, \
Dconv_grad_filter_ops.cc570 int64 padding_left = -1, padding_right = -1; in operator ()() local
574 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', &padding_left, in operator ()()
587 col_dilation, col_stride, padding, &expected_out_cols, &padding_left, 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 ()()
694 const int64 input_pad_left = padding_left - common_padding_cols; in operator ()()
940 const std::array<int, 2>& padding_left, \
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.cc73 memory::dims padding_left; member
85 memory::dims padding_left, memory::dims padding_right) in MklConvFwdParams()
92 padding_left(padding_left), in MklConvFwdParams()
234 convFwdDims.strides, convFwdDims.dilations, convFwdDims.padding_left, in Setup()
240 convFwdDims.dilations, convFwdDims.padding_left, in Setup()
371 key_creator.AddAsKey(convFwdDims.padding_left); 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()
966 &dilations, &dst_dims_tf_order, &dst_dims_mkl_order, &padding_left, in Compute()
1074 dst_dims_mkl_order, strides, dilations, padding_left, padding_right); in Compute()
[all …]
Dconv_2d_gpu.h366 Dimension<NDIMS - 2> padding_left) {
376 input_tensor_index[i] = output_tensor_index[i] - padding_left[i - 1];
395 Dimension<NDIMS - 2> padding_left) {
406 input_tensor_index[i] = output_tensor_index[i] - padding_left[i - 2];
476 const std::array<int, NDIMS - 2>& padding_left,
490 const Dimension<NDIMS - 2> padding_left_dim(padding_left);
Dmkl_pooling_ops_common.cc66 fwdParams.padding_left, fwdParams.padding_right, padding_kind::zero)); in Setup()
152 bwdParams.strides, bwdParams.filter_dims, bwdParams.padding_left, in Setup()
160 bwdParams.padding_left, bwdParams.padding_right, padding_kind::zero)); in Setup()
Dconv_2d.h250 const std::array<int, NDIMS - 2>& padding_left,
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),
118 DenseIndex padding_left() const { return m_padding_left; }
222 …m_outputCols = numext::ceil((m_input_cols_eff + op.padding_left() + op.padding_right() - m_patch_c…
224 m_colPaddingLeft = op.padding_left();
DTensorVolumePatch.h81 … DenseIndex padding_left, DenseIndex padding_right,
88 m_padding_left(padding_left), m_padding_right(padding_right),
126 DenseIndex padding_left() const { return m_padding_left; }
232 …m_outputCols = numext::ceil((m_input_cols_eff + op.padding_left() + op.padding_right() - m_patch_c…
235 m_colPaddingLeft = op.padding_left();
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h432 const int padding_left = function_params.padding_left;
438 padding_left > 0 && width_block_number == 0;
626 const int padding_left = function_params->padding_left;
632 padding_left > 0 && width_block_number == 0;
899 const int padding_left = function_params->padding_left;
909 padding_left > 0 && width_block_number == 0;
1202 TFLITE_DCHECK_EQ(function_params->padding_left, 0);
1405 const int padding_left = function_params->padding_left;
1423 padding_left > 0 && width_block_number == 0;
1662 const int padding_left = function_params->padding_left;
[all …]
Ddepthwiseconv_uint8_3x3_filter.h396 int32 padding_left;
3919 TFLITE_DCHECK_EQ(function_params->padding_left, 0);
4118 const int padding_left = function_params->padding_left;
4136 padding_left > 0 && width_block_number == 0;
4371 const int padding_left = function_params->padding_left;
4381 padding_left > 0 && width_block_number == 0;
4724 TFLITE_DCHECK_EQ(function_params->padding_left, 0);
7142 const int padding_left = params.padding_values.width;
7145 std::max(padded_width - input_width - padding_left, 0);
7151 function_params.padding_left = padding_left;
[all …]

12