Home
last modified time | relevance | path

Searched refs:kernel_stride (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc63 const Array3D<float>& lhs, const Array3D<float>& rhs, int64 kernel_stride, in ConvArray3D() argument
66 lhs, rhs, kernel_stride, padding, 1, 1, in ConvArray3D()
72 const Array3D<float>& lhs, const Array3D<float>& rhs, int64 kernel_stride, in ConvArray3DGeneralDimensionsDilated() argument
97 a4dlhs, a4drhs, {kernel_stride, 1}, padding, {lhs_dilation, 1}, in ConvArray3DGeneralDimensionsDilated()
111 std::pair<int64, int64> kernel_stride, Padding padding) { in ConvArray4D() argument
113 lhs, rhs, kernel_stride, padding, in ConvArray4D()
121 std::pair<int64, int64> kernel_stride, in SeparableConvArray4D() argument
146 return ConvArray4D(input, weights, kernel_stride, padding); in SeparableConvArray4D()
467 std::pair<int64, int64> kernel_stride, Padding padding, in ConvArray4DGeneralDimensions() argument
469 return ConvArray4DGeneralDimensionsDilated(lhs, rhs, kernel_stride, padding, in ConvArray4DGeneralDimensions()
[all …]
Dreference_util.h73 std::pair<int64, int64> kernel_stride, Padding padding);
79 std::pair<int64, int64> kernel_stride, Padding padding,
86 std::pair<int64, int64> kernel_stride, Padding padding,
95 int64 kernel_stride,
100 const Array3D<float>& lhs, const Array3D<float>& rhs, int64 kernel_stride,
111 std::pair<int64, int64> kernel_stride, Padding padding);
/external/tensorflow/tensorflow/core/kernels/
Dmkl_pooling_ops_common.cc429 mkl_params->kernel_stride[0] = params.col_stride; in ExtractMklOpParams()
430 mkl_params->kernel_stride[1] = params.row_stride; in ExtractMklOpParams()
Dmkl_maxpooling_op.cc171 params.kernel_size, params.kernel_stride, params.in_offset, in MklCreateLayoutsAndPrimitives()
350 params.kernel_size, params.kernel_stride, params.in_offset, in MklCreatePrimitives()
357 params.kernel_size, params.kernel_stride, params.in_offset, in MklCreatePrimitives()
Dmkl_avgpooling_op.cc179 params.kernel_size, params.kernel_stride, params.in_offset, in MklCreateLayoutsAndPrimitives()
399 params.kernel_size, params.kernel_stride, params.in_offset, in MklCreateLayoutsAndPrimitives()
Dmkl_pooling_ops_common.h765 size_t kernel_stride[2]; member