Searched defs:new_dim_numbers (Results 1 – 4 of 4) sorted by relevance
72 DotDimensionNumbers new_dim_numbers = dim_numbers; in ElideDegenerateBatchDimensionFromBatchDot() local
112 DotDimensionNumbers new_dim_numbers = dot->dot_dimension_numbers(); in FoldTransposeIntoDot() local
486 ConvolutionDimensionNumbers new_dim_numbers = dim_numbers; in BringSpaceNextToBatch() local1904 auto new_dim_numbers = permuted_conv_dims_numbers; in PropagateOnConv() local2397 auto new_dim_numbers = permuted_conv_dims_numbers; in PropagateOnBackpropFilterConv() local2743 auto new_dim_numbers = dim_numbers; in PerformSpaceToBatchOnConvolution() local
1046 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedLhs() local1082 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedRhs() local