Searched refs:first_dim_to_split (Results 1 – 1 of 1) sorted by relevance
3182 const int64_t first_dim_to_split = GetFirstChosenSpatialDim(convolution); in PropagateOnBackpropFilterConv() local3185 permuted_conv_dims_numbers.input_spatial_dimensions(first_dim_to_split + in PropagateOnBackpropFilterConv()