Searched refs:PermuteBatchAndOffsetDims (Results 1 – 1 of 1) sorted by relevance
252 static StatusOr<HloInstruction*> PermuteBatchAndOffsetDims( in PermuteBatchAndOffsetDims() function367 return PermuteBatchAndOffsetDims(accumulator_with_batch_dims_decanonicalized, in ExpandInstruction()