Searched refs:permuted_conv_dims_numbers (Results 1 – 1 of 1) sorted by relevance
1859 auto permuted_conv_dims_numbers = original_conv_dims; in PropagateOnConv() local1865 permuted_conv_dims_numbers.set_input_batch_dimension(activations_batch_dim); in PropagateOnConv()1866 permuted_conv_dims_numbers.set_input_feature_dimension( in PropagateOnConv()1871 permuted_conv_dims_numbers.set_input_spatial_dimensions( in PropagateOnConv()1881 GetConvolutionDetails(convolution, permuted_conv_dims_numbers); in PropagateOnConv()1890 activations_new, permuted_conv_dims_numbers, in PropagateOnConv()1904 auto new_dim_numbers = permuted_conv_dims_numbers; in PropagateOnConv()1910 permuted_conv_dims_numbers.output_spatial_dimensions( in PropagateOnConv()1989 j < permuted_conv_dims_numbers.output_spatial_dimensions_size(); ++j) { in PropagateOnConv()1991 dim_map[permuted_conv_dims_numbers.output_batch_dimension()] = dim_count; in PropagateOnConv()[all …]