Searched defs:new_dim_numbers (Results 1 – 3 of 3) sorted by relevance
71 DotDimensionNumbers new_dim_numbers = dim_numbers; in ElideDegenerateBatchDimensionFromBatchDot() local
1050 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedLhs() local1086 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedRhs() local
656 ConvolutionDimensionNumbers new_dim_numbers = dim_numbers; in BringSpaceNextToBatch() local2594 auto new_dim_numbers = permuted_conv_dims_numbers; in PropagateOnConv() local3270 auto new_dim_numbers = permuted_conv_dims_numbers; in PropagateOnBackpropFilterConv() local3748 auto new_dim_numbers = dim_numbers; in PerformSpaceToBatchOnConvolution() local