Home
last modified time | relevance | path

Searched refs:padding_bottom (Results 1 – 20 of 20) sorted by relevance

/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()
48 int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
55 padding_top, padding_bottom, padding_left, padding_right, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc32 int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, 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.cc57 int64 col_stride, int64 padding_top, int64 padding_bottom, in MKLConvImpl() argument
80 memory::dims conv1_padding_r = {ToInt(padding_bottom), ToInt(padding_right)}; in MKLConvImpl()
157 int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, 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.h33 tensorflow::int64 padding_top, tensorflow::int64 padding_bottom,
47 tensorflow::int64 padding_bottom, tensorflow::int64 padding_left,
Druntime_conv2d.h33 tensorflow::int64 padding_bottom, tensorflow::int64 padding_left,
47 tensorflow::int64 padding_top, tensorflow::int64 padding_bottom,
Druntime_conv2d_impl.h34 int64 padding_bottom, int64 padding_left, in EigenConvImpl() argument
79 padding_bottom, static_cast<ScalarType>(0.0f)) in EigenConvImpl()
Druntime_conv2d_mkl.h33 tensorflow::int64 padding_bottom, tensorflow::int64 padding_left,
Dir_emitter.cc1164 int64 padding_bottom = window.dimensions(0).padding_high(); in HandleConvolution() local
1234 b_.getInt64(padding_bottom), in HandleConvolution()
/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h158 const TensorIndex padding_bottom = inputRows - (outputRows - 1) * row_stride - variable
165 eigen_assert(padding_bottom >= 0);
282 padding_bottom, padding_left, padding_right, OutScalar(0))
289 padding_top, padding_bottom, padding_left,
414 const TensorIndex padding_bottom = padRows - padding_top; variable
486 padding_bottom, padding_left, padding_right, OutScalar(0))
493 padding_top, padding_bottom, padding_left,
Dconv_grad_input_ops.cc708 int64 padding_top = -1, padding_bottom = -1; in operator ()() local
712 &padding_bottom); in operator ()()
722 &padding_bottom)); in operator ()()
810 const int64 common_padding_rows = std::min(padding_top, padding_bottom); in operator ()()
813 if (padding_top != padding_bottom || padding_left != padding_right) { in operator ()()
817 const int64 padding_rows_diff = std::abs(padding_bottom - padding_top); in operator ()()
1010 if (padding_top != padding_bottom || padding_left != padding_right) { in operator ()()
1024 const int64 input_pad_bottom = padding_bottom - common_padding_rows; in operator ()()
Dconv_ops.cc671 int64 padding_top = -1, padding_bottom = -1; in operator ()() local
675 &padding_bottom); in operator ()()
682 &padding_top, &padding_bottom); in operator ()()
693 const int64 common_padding_rows = std::min(padding_top, padding_bottom); in operator ()()
695 if (padding_top != padding_bottom || padding_left != padding_right) { in operator ()()
706 const int64 padding_rows_diff = std::abs(padding_bottom - padding_top); in operator ()()
717 const int64 input_pad_bottom = padding_bottom - common_padding_rows; in operator ()()
Dconv_grad_filter_ops.cc569 int64 padding_top = -1, padding_bottom = -1; in operator ()() local
573 &padding_bottom); in operator ()()
583 &padding_bottom)); in operator ()()
679 const int64 common_padding_rows = std::min(padding_top, padding_bottom); in operator ()()
682 if (padding_top != padding_bottom || padding_left != padding_right) { in operator ()()
686 const int64 padding_rows_diff = std::abs(padding_bottom - padding_top); in operator ()()
693 const int64 input_pad_bottom = padding_bottom - common_padding_rows; in operator ()()
Deigen_backward_cuboid_convolutions.h158 const TensorIndex padding_bottom = inputRows - (outputRows - 1) * row_stride - variable
167 eigen_assert(padding_bottom >= 0);
284 padding_bottom_z, padding_top, padding_bottom,
293 padding_bottom, padding_left, padding_right)
Deigen_volume_patch.h106 static_cast<Index>(op.padding_top() + op.padding_bottom()) - in CustomTensorEvaluator()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorImagePatch.h84 … DenseIndex padding_top, DenseIndex padding_bottom,
91 m_padding_explicit(true), m_padding_top(padding_top), m_padding_bottom(padding_bottom),
116 DenseIndex padding_bottom() const { return m_padding_bottom; }
221 …m_outputRows = numext::ceil((m_input_rows_eff + op.padding_top() + op.padding_bottom() - m_patch_r…
DTensorVolumePatch.h80 … DenseIndex padding_top, DenseIndex padding_bottom,
87 …m_padding_bottom_z(padding_bottom_z), m_padding_top(padding_top), m_padding_bottom(padding_bottom),
124 DenseIndex padding_bottom() const { return m_padding_bottom; }
231 …m_outputRows = numext::ceil((m_input_rows_eff + op.padding_top() + op.padding_bottom() - m_patch_r…
DTensorBase.h702 const Index padding_top, const Index padding_bottom, in extract_image_patches() argument
707 … padding_top, padding_bottom, padding_left, padding_right, padding_value); in extract_image_patches()
725 const Index padding_top, const Index padding_bottom,
727 …e, col_inflate_stride, padding_top_z, padding_bottom_z, padding_top, padding_bottom, padding_left,…
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h435 const int padding_bottom = function_params.padding_bottom;
445 padding_bottom > 0 &&
629 const int padding_bottom = function_params->padding_bottom;
639 padding_bottom > 0 &&
902 const int padding_bottom = function_params->padding_bottom;
916 padding_bottom > 0 &&
1200 TFLITE_DCHECK_EQ(function_params->padding_bottom, 0);
1408 const int padding_bottom = function_params->padding_bottom;
1430 padding_bottom > 0 &&
1665 const int padding_bottom = function_params->padding_bottom;
[all …]
Ddepthwiseconv_uint8_3x3_filter.h399 int32 padding_bottom;
3917 TFLITE_DCHECK_EQ(function_params->padding_bottom, 0);
4121 const int padding_bottom = function_params->padding_bottom;
4143 padding_bottom > 0 &&
4374 const int padding_bottom = function_params->padding_bottom;
4388 padding_bottom > 0 &&
4727 TFLITE_DCHECK_EQ(function_params->padding_bottom, 0);
7148 const int padding_bottom =
7154 function_params.padding_bottom = padding_bottom;
7157 TFLITE_DCHECK_LE(padding_top, padding_bottom);
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc111 int64 padding_bottom = 2; in SimpleConv2DBackpropInput() local
127 &output_height, &padding_top, &padding_bottom)); in SimpleConv2DBackpropInput()
159 static_cast<int>(padding_bottom), in SimpleConv2DBackpropInput()