提交 a262b3b0 authored 作者: Frederic Bastien's avatar Frederic Bastien

Make sure the docstring is attached to the right object and reuse objects.

上级 26d0f24a
...@@ -217,10 +217,10 @@ class GpuBatchedDot(GpuOp): ...@@ -217,10 +217,10 @@ 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.
""" """
BatchedDotOp = batched_dot
class GpuDot22(GpuOp): class GpuDot22(GpuOp):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论