| /external/tensorflow/tensorflow/compiler/xla/ |
| reference_util.cc | 93 const Array3D<float>& lhs, const Array3D<float>& rhs, int64 kernel_stride,
96 lhs, rhs, kernel_stride, padding, 1, 1,
102 const Array3D<float>& lhs, const Array3D<float>& rhs, int64 kernel_stride,
129 a4dlhs, a4drhs, {kernel_stride, 1}, padding, {lhs_dilation, 1},
144 std::pair<int64, int64> kernel_stride, Padding padding) {
146 lhs, rhs, kernel_stride, padding,
154 std::pair<int64, int64> kernel_stride,
179 return ConvArray4D(input, weights, kernel_stride, padding);
499 std::pair<int64, int64> kernel_stride, Padding padding,
501 return ConvArray4DGeneralDimensionsDilated(lhs, rhs, kernel_stride, padding
[all...] |
| reference_util.h | 60 std::pair<int64, int64> kernel_stride, Padding padding);
66 std::pair<int64, int64> kernel_stride, Padding padding,
73 std::pair<int64, int64> kernel_stride, Padding padding,
82 int64 kernel_stride,
87 const Array3D<float>& lhs, const Array3D<float>& rhs, int64 kernel_stride,
92 // kernel_stride and padding applies to the depthwise convolution during
98 std::pair<int64, int64> kernel_stride, Padding padding);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_pooling_ops_common.cc | 175 mkl_params->kernel_stride[0] = params.col_stride;
176 mkl_params->kernel_stride[1] = params.row_stride;
|
| mkl_maxpooling_op.cc | 170 params.kernel_size, params.kernel_stride, params.in_offset,
348 params.kernel_size, params.kernel_stride, params.in_offset,
355 params.kernel_size, params.kernel_stride, params.in_offset,
[all...] |
| mkl_pooling_ops_common.h | 410 size_t kernel_stride[2]; member in struct:tensorflow::__anon39849
|
| mkl_avgpooling_op.cc | 179 params.kernel_size, params.kernel_stride, params.in_offset,
399 params.kernel_size, params.kernel_stride, params.in_offset,
|
