Searched defs:kernel_stride (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | reference_util.cc | 48 const Array3D<float>& lhs, const Array3D<float>& rhs, int64_t kernel_stride, in ConvArray3D()
57 const Array3D<float>& lhs, const Array3D<float>& rhs, int64_t kernel_stride, in ConvArray3DGeneralDimensionsDilated()
96 std::pair<int64_t, int64_t> kernel_stride, Padding padding) { in ConvArray4D() argument
106 std::pair<int64_t, int64_t> kernel_stride, in SeparableConvArray4D() argument
403 std::pair<int64_t, int64_t> kernel_stride, Padding padding, in ConvArray4DGeneralDimensions() argument
413 std::pair<int64_t, int64_t> kernel_stride, Padding padding, in ConvArray4DGeneralDimensionsDilated() argument
|