提交 39f85b45 authored 作者: Olivier Delalleau's avatar Olivier Delalleau 提交者: Frederic

A few typo fixes

上级 72d2d3d2
...@@ -383,7 +383,7 @@ pp = pprint ...@@ -383,7 +383,7 @@ pp = pprint
# colors not used: orange, ambre#FFBF00, purple, pink, # colors not used: orange, ambre#FFBF00, purple, pink,
# used by default: green, bug, grey, red # used by default: green, blue, grey, red
default_colorCodes = {'GpuFromHost' : 'red', default_colorCodes = {'GpuFromHost' : 'red',
'HostFromGpu' : 'red', 'HostFromGpu' : 'red',
'Scan' : 'yellow', 'Scan' : 'yellow',
...@@ -428,17 +428,16 @@ def pydotprint(fct, outfile=None, ...@@ -428,17 +428,16 @@ def pydotprint(fct, outfile=None,
we will print only the variable name. we will print only the variable name.
Otherwise, we concatenate the type to the var name. Otherwise, we concatenate the type to the var name.
In the graph, ellipse are an Apply Node(the execution of an op) and box are variable. In the graph, ellipses are Apply Nodes (the execution of an op) and boxes are variables.
If variable have name they are used as the text(if multiple var have the same name, they will be merged in the graph). If variables have names they are used as text (if multiple vars have the same name, they will be merged in the graph).
Otherwise, if the variable is constant, we print the value and finaly we print the type + an uniq number to don't have multiple var merged. Otherwise, if the variable is constant, we print its value and finally we print the type + a unique number to prevent multiple vars from being merged.
We print the op of the apply in the Apply box with a number that represent the toposort order of application of those Apply. We print the op of the apply in the Apply box with a number that represents the toposort order of application of those Apply.
If an Apply have more then 1 input, print add a label to the edge that in the index of the inputs. If an Apply has more than 1 input, we label each edge between an input and the Apply node with the input's index.
green box are inputs to the graph green boxes are inputs to the graph
blue box are outputs of the graph blue boxes are outputs of the graph
grey box are var generated by the graph that are not output and are not used. grey boxes are vars generated by the graph that are not outputs and are not used
red ellipses are transfer to/from the gpu. red ellipses are transfers from/to the gpu (ops with names GpuFromHost, HostFromGpu)
op with those name GpuFromHost, HostFromGpu
""" """
if colorCodes is None: if colorCodes is None:
colorCodes = default_colorCodes colorCodes = default_colorCodes
......
...@@ -36,13 +36,15 @@ def test_pydotprint_cond_highlight(): ...@@ -36,13 +36,15 @@ def test_pydotprint_cond_highlight():
assert s.getvalue() == 'pydotprint: cond_highlight is set but there is no IfElse node in the graph\n' assert s.getvalue() == 'pydotprint: cond_highlight is set but there is no IfElse node in the graph\n'
def test_pydotprint_profile(): def test_pydotprint_profile():
""" Just check that pydotprint don't crash with ProfileMode""" """Just check that pydotprint does not crash with ProfileMode."""
A = tensor.matrix() A = tensor.matrix()
f = theano.function([A],A+1,mode='ProfileMode') f = theano.function([A], A+1, mode='ProfileMode')
theano.printing.pydotprint(f, print_output_file=False) theano.printing.pydotprint(f, print_output_file=False)
def test_min_informative_str(): def test_min_informative_str():
""" evaluates a reference output to make sure the """ evaluates a reference output to make sure the
min_informative_str function works as intended """ min_informative_str function works as intended """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论