提交 ed3034a3 authored 作者: Ramana.S's avatar Ramana.S

added BatchedDotOp

上级 72d3a280
...@@ -217,6 +217,7 @@ class GpuBatchedDot(GpuOp): ...@@ -217,6 +217,7 @@ class GpuBatchedDot(GpuOp):
return [xshp[:-1] + yshp[2:]] return [xshp[:-1] + yshp[2:]]
batched_dot = GpuBatchedDot() batched_dot = GpuBatchedDot()
BatchedDotOp = GpuBatchedDot()
""" """
Call cublasSgemmBatched. Take 2 3d tensor as input. Call cublasSgemmBatched. Take 2 3d tensor as input.
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论