Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc48 const Array3D<float>& lhs, const Array3D<float>& rhs, int64 kernel_stride, in ConvArray3D() argument
51 lhs, rhs, kernel_stride, padding, 1, 1, in ConvArray3D()
57 const Array3D<float>& lhs, const Array3D<float>& rhs, int64 kernel_stride, in ConvArray3DGeneralDimensionsDilated() argument
82 a4dlhs, a4drhs, {kernel_stride, 1}, padding, {lhs_dilation, 1}, in ConvArray3DGeneralDimensionsDilated()
96 std::pair<int64, int64> kernel_stride, Padding padding) { in ConvArray4D() argument
98 lhs, rhs, kernel_stride, padding, in ConvArray4D()
106 std::pair<int64, int64> kernel_stride, in SeparableConvArray4D() argument
131 return ConvArray4D(input, weights, kernel_stride, padding); in SeparableConvArray4D()
401 std::pair<int64, int64> kernel_stride, Padding padding, in ConvArray4DGeneralDimensions() argument
403 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);