Searched refs:conv_dnums (Results 1 – 3 of 3) sorted by relevance
274 ConvolutionDimensionNumbers conv_dnums; in TEST_F() local275 conv_dnums.set_input_batch_dimension(0); in TEST_F()276 conv_dnums.set_output_batch_dimension(0); in TEST_F()277 conv_dnums.set_input_feature_dimension(1); in TEST_F()278 conv_dnums.set_output_feature_dimension(1); in TEST_F()279 conv_dnums.add_input_spatial_dimensions(2); in TEST_F()280 conv_dnums.add_output_spatial_dimensions(2); in TEST_F()281 conv_dnums.add_input_spatial_dimensions(3); in TEST_F()282 conv_dnums.add_output_spatial_dimensions(3); in TEST_F()283 conv_dnums.set_kernel_input_feature_dimension(0); in TEST_F()[all …]
116 const ConvolutionDimensionNumbers& conv_dnums = in MatchBackwardFilter() local118 auto input_batch_dim = conv_dnums.input_batch_dimension(); in MatchBackwardFilter()119 auto input_feature_dim = conv_dnums.input_feature_dimension(); in MatchBackwardFilter()120 auto input_spatial_dims = conv_dnums.input_spatial_dimensions(); in MatchBackwardFilter()121 auto kernel_input_feature_dim = conv_dnums.kernel_input_feature_dimension(); in MatchBackwardFilter()122 auto kernel_output_feature_dim = conv_dnums.kernel_output_feature_dimension(); in MatchBackwardFilter()123 auto kernel_spatial_dims = conv_dnums.kernel_spatial_dimensions(); in MatchBackwardFilter()124 auto output_batch_dim = conv_dnums.output_batch_dimension(); in MatchBackwardFilter()125 auto output_feature_dim = conv_dnums.output_feature_dimension(); in MatchBackwardFilter()126 auto output_spatial_dims = conv_dnums.output_spatial_dimensions(); in MatchBackwardFilter()
49 Window* conv_window, const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedAndSlicedInput() argument64 for (size_t i = 0; i < conv_dnums.input_spatial_dimensions().size(); ++i) { in MaybePaddedAndSlicedInput()65 int64 dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput()98 for (size_t i = 0; i < conv_dnums.input_spatial_dimensions().size(); ++i) { in MaybePaddedAndSlicedInput()99 int64 dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput()124 const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedKernel() argument135 for (size_t i = 0; i < conv_dnums.kernel_spatial_dimensions().size(); ++i) { in MaybePaddedKernel()136 int64 dim = conv_dnums.kernel_spatial_dimensions(i); in MaybePaddedKernel()