Searched defs:batch_dot (Results 1 – 3 of 3) sorted by relevance
25 HloInstruction* batch_dot) { in ElideDegenerateBatchDimensionFromBatchDot()
1305 auto batch_dot = BatchDot(x, y); in XLA_TEST_P() local
2022 def batch_dot(x, y, axes=None): function