Searched refs:dnums2d (Results 1 – 1 of 1) sorted by relevance
77 ConvolutionDimensionNumbers dnums2d = dnums; in ConvArray3DGeneralDimensionsDilated() local78 dnums2d.add_input_spatial_dimensions(3); in ConvArray3DGeneralDimensionsDilated()79 dnums2d.add_kernel_spatial_dimensions(3); in ConvArray3DGeneralDimensionsDilated()80 dnums2d.add_output_spatial_dimensions(3); in ConvArray3DGeneralDimensionsDilated()83 {rhs_dilation, 1}, dnums2d); in ConvArray3DGeneralDimensionsDilated()