提交 7061380c authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Rename function to reflect changes.

上级 4af1e536
......@@ -55,12 +55,12 @@ def numpy_maxandargmax(X, axis=None):
return (ref_max, np.argmax(reshaped_x, axis=-1))
def check_if_gpu_maxandargmax_in_graph(theano_function):
def check_if_gpu_reduce_in_graph(theano_function):
assert any(isinstance(node.op, (GpuMaxAndArgmax, GpuDnnReduction))
for node in theano_function.maker.fgraph.apply_nodes)
def check_if_gpu_maxandargmax_not_in_graph(theano_function):
def check_if_gpu_reduce_not_in_graph(theano_function):
assert all(not isinstance(node.op, (GpuMaxAndArgmax, GpuDnnReduction))
for node in theano_function.maker.fgraph.apply_nodes)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论