Searched defs:batch_dot (Results 1 – 2 of 2) sorted by relevance
25 HloInstruction* batch_dot) { in ElideDegenerateBatchDimensionFromBatchDot()
1482 def batch_dot(x, y, axes=None): function