| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_single_threaded_conv2d.cc | 31 int64 padding_bottom, int64 padding_left, int64 padding_right,
38 padding_top, padding_bottom, padding_left, padding_right, local
49 int64 padding_left, int64 padding_right, int64 lhs_row_dilation,
55 padding_top, padding_bottom, padding_left, padding_right, local
|
| ir_emitter.cc | 1165 int64 padding_left = local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| eigen_backward_spatial_convolutions.h | 156 const TensorIndex padding_left = kernelColsEff - 1 - forward_pad_left; local
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; local
416 const TensorIndex padding_right = padCols - padding_left;
486 padding_bottom, padding_left, padding_right, OutScalar(0))
493 padding_top, padding_bottom, padding_left,
|
| eigen_backward_cuboid_convolutions.h | 153 const TensorIndex padding_left = kernelColsEff - 1 - forward_pad_left; local
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; local
477 const auto left_pad_cols = kernelCols - 1 - padding_left;
|
| mkl_avgpooling_op.cc | 477 memory::dims filter_dims, strides, padding_left, padding_right; variable
480 &padding_left, &padding_right, is_pool2d);
505 strides, padding_left, padding_right,
599 memory::dims filter_dims, strides, padding_left, padding_right; variable
601 &padding_left, &padding_right, is_pool2d);
625 strides, padding_left, padding_right,
|
| mkl_conv_grad_filter_ops.cc | 58 memory::dims padding_left; member in struct:tensorflow::MklConvBwdFilterParams
65 memory::dims dilations, memory::dims padding_left,
73 padding_left(padding_left),
232 convBwdFilterDims.padding_left, convBwdFilterDims.padding_right,
238 convBwdFilterDims.dilations, convBwdFilterDims.padding_left,
247 convBwdFilterDims.padding_left, convBwdFilterDims.padding_right,
342 key_creator.AddAsKey(convBwdFilterDims.padding_left);
425 memory::dims padding_left, padding_right, dilations, strides, local
435 &padding_left, &padding_right, false, is_depthwise)
[all...] |
| mkl_conv_grad_input_ops.cc | 60 memory::dims padding_left; member in struct:tensorflow::MklConvBwdInputParams
66 memory::dims dilations, memory::dims padding_left,
73 padding_left(padding_left),
190 convBwdInputDims.dilations, convBwdInputDims.padding_left,
196 convBwdInputDims.dilations, convBwdInputDims.padding_left,
281 key_creator.AddAsKey(convBwdInputDims.padding_left);
362 memory::dims padding_left, padding_right, dilations, strides; local
371 &padding_left, &padding_right, false, is_depthwise);
404 padding_left, padding_right
[all...] |
| conv_grad_filter_ops.cc | 570 int64 padding_left = -1, padding_right = -1; local
574 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', &padding_left,
587 col_dilation, col_stride, padding, &expected_out_cols, &padding_left,
680 const int64 common_padding_cols = std::min(padding_left, padding_right);
682 if (padding_top != padding_bottom || padding_left != padding_right) {
687 const int64 padding_cols_diff = std::abs(padding_right - padding_left);
694 const int64 input_pad_left = padding_left - common_padding_cols;
[all...] |
| conv_grad_input_ops.cc | 709 int64 padding_left = -1, padding_right = -1; local
713 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', &padding_left,
726 col_dilation, col_stride, padding, &expected_out_cols, &padding_left,
[all...] |
| conv_ops.cc | 672 int64 padding_left = -1, padding_right = -1; local
676 GetExplicitPaddingForDim(explicit_paddings, data_format, 'W', &padding_left,
689 &padding_left, &padding_right);
694 const int64 common_padding_cols = std::min(padding_left, padding_right);
695 if (padding_top != padding_bottom || padding_left != padding_right) {
707 const int64 padding_cols_diff = std::abs(padding_right - padding_left);
718 const int64 input_pad_left = padding_left - common_padding_cols;
[all...] |
| mkl_maxpooling_op.cc | 568 memory::dims filter_dims, strides, padding_left, padding_right; variable
570 &padding_left, &padding_right, is_pool2d);
582 strides, padding_left, padding_right,
710 memory::dims filter_dims, strides, padding_left, padding_right; variable
712 &padding_left, &padding_right, is_pool2d);
735 strides, padding_left, padding_right, algorithm::pooling_max,
[all...] |
| mkl_pooling_ops_common.h | 50 memory::dims padding_left; member in struct:tensorflow::MklPoolingParams
57 memory::dims padding_left, memory::dims padding_right,
63 padding_left(padding_left),
192 key_creator.AddAsKey(fwdParams.padding_left);
344 key_creator.AddAsKey(bwdParams.padding_left);
523 memory::dims* padding_left, memory::dims* padding_right,
531 *padding_left = memory::dims({static_cast<int>(pool_params->pad_top),
544 *padding_left = memory::dims({static_cast<int>(pool_params->pad_P1),
|