Home
last modified time | relevance | path

Searched refs:graph_scope (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/tests/st/graph_kernel/model/
Dtest_split.py59 with gb.graph_scope("main"):
71 with gb.graph_scope("main"):
85 with gb.graph_scope("main"):
99 with gb.graph_scope("main"):
117 with gb.graph_scope("main") as g:
135 with gb.graph_scope("main"):
150 with gb.graph_scope("main"):
163 with gb.graph_scope("main"):
175 with gb.graph_scope("main"):
188 with gb.graph_scope("main"):
[all …]
Dtest_graph_parallel.py20 with gb.graph_scope('injective') as _:
29 with gb.graph_scope('reduce') as _:
/third_party/mindspore/mindspore/_extends/graph_kernel/expanders/
D_utils.py44 with graph_builder.graph_scope(self.name) as graph_scope:
47 graph_scope.set_input(*self.inputs)
51 graph_scope.set_output(*outputs)
54 graph_scope.set_output(outputs)
/third_party/mindspore/mindspore/_extends/graph_kernel/model/
Dmodel_builder.py52 def graph_scope(self, name): member in GraphBuilder
142 with builder.graph_scope(desc['op']):