Searched refs:dnums2d (Results 1 – 1 of 1) sorted by relevance
92 ConvolutionDimensionNumbers dnums2d = dnums; in ConvArray3DGeneralDimensionsDilated() local93 dnums2d.add_input_spatial_dimensions(3); in ConvArray3DGeneralDimensionsDilated()94 dnums2d.add_kernel_spatial_dimensions(3); in ConvArray3DGeneralDimensionsDilated()95 dnums2d.add_output_spatial_dimensions(3); in ConvArray3DGeneralDimensionsDilated()98 {rhs_dilation, 1}, dnums2d); in ConvArray3DGeneralDimensionsDilated()