Home
last modified time | relevance | path

Searched refs:graph_kernel_flags (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_low_precision.py93 …context.set_context(graph_kernel_flags="--enable_low_precision=true --disable_pass=highlevelopt2.a…
101 context.set_context(graph_kernel_flags="--enable_low_precision=true")
110 …context.set_context(graph_kernel_flags="--enable_low_precision=true --disable_pass=highlevelopt2.a…
119 context.set_context(graph_kernel_flags="--enable_low_precision=true")
Dtest_layernorm_stitch.py66 context.set_context(graph_kernel_flags="--enable_stitch_fusion=true")
Dtest_dropoutgrad_reducesum_stitch.py69 context.set_context(graph_kernel_flags="--enable_stitch_fusion=true")
Dtest_softmax_stitch.py73 context.set_context(graph_kernel_flags="--enable_stitch_fusion=true")
/third_party/mindspore/mindspore/ccsrc/utils/context/
Dgraph_kernel_flags.h195 GraphKernelFlags(const std::string &graph_kernel_flags, bool enable_graph_kernel) in GraphKernelFlags() argument
196 : flags_cache_(graph_kernel_flags), enable_graph_kernel_(enable_graph_kernel) {} in GraphKernelFlags()
/third_party/mindspore/mindspore/
Dcontext.py520 env_config_path=str, graph_kernel_flags=str, save_compile_cache=bool,