Searched refs:original_conv_dims (Results 1 – 1 of 1) sorted by relevance
1852 auto original_conv_dims = convolution->convolution_dimension_numbers(); in PropagateOnConv() local1854 const int64 old_space_dim = original_conv_dims.input_spatial_dimensions( in PropagateOnConv()1859 auto permuted_conv_dims_numbers = original_conv_dims; in PropagateOnConv()1862 DimLookUp(permute_dims, original_conv_dims.input_batch_dimension()); in PropagateOnConv()1864 DimLookUp(permute_dims, original_conv_dims.input_feature_dimension()); in PropagateOnConv()1869 for (int64 i = 0; i < original_conv_dims.input_spatial_dimensions_size(); in PropagateOnConv()1873 original_conv_dims.input_spatial_dimensions(i))); in PropagateOnConv()1876 const int64 old_batch_dim = original_conv_dims.input_batch_dimension(); in PropagateOnConv()2027 std::make_pair(original_conv_dims.output_batch_dimension(), in PropagateOnConv()2028 original_conv_dims.output_spatial_dimensions( in PropagateOnConv()[all …]