提交 3226ffd2 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix wrong spelling of _f16_ok

上级 33b437b7
......@@ -215,7 +215,7 @@ class Shape(gof.Op):
@note: Non-differentiable.
"""
_fp16_ok = True
_f16_ok = True
# Mapping from Type to C code (and version) to use.
# In the C code, the name of the input variable is %(iname)s,
......@@ -310,7 +310,7 @@ class Shape_i(gof.Op):
@note: Non-differentiable.
"""
_fp16_ok = True
_f16_ok = True
# Mapping from Type to C code (and version) to use.
# In the C code, the name of the input variable is %(iname)s,
......
......@@ -21,7 +21,7 @@ from .comp import NVCC_compiler
class GpuSubtensor(HideC, Subtensor):
_fp16_ok = True
_f16_ok = True
def make_node(self, x, *inputs):
rval = tensor.Subtensor.make_node(self, x, *inputs)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论