Searched refs:original_conv_dims (Results 1 – 1 of 1) sorted by relevance
2543 auto original_conv_dims = convolution->convolution_dimension_numbers(); in PropagateOnConv() local2549 auto permuted_conv_dims_numbers = original_conv_dims; in PropagateOnConv()2552 DimLookUp(permute_dims, original_conv_dims.input_batch_dimension()); in PropagateOnConv()2554 DimLookUp(permute_dims, original_conv_dims.input_feature_dimension()); in PropagateOnConv()2559 for (int64_t i = 0; i < original_conv_dims.input_spatial_dimensions_size(); in PropagateOnConv()2563 original_conv_dims.input_spatial_dimensions(i))); in PropagateOnConv()2566 const int64_t old_batch_dim = original_conv_dims.input_batch_dimension(); in PropagateOnConv()2727 original_conv_dims.output_batch_dimension(); in PropagateOnConv()2729 original_conv_dims.output_feature_dimension(); in PropagateOnConv()2731 original_conv_dims.output_spatial_dimensions( in PropagateOnConv()[all …]