Home
last modified time | relevance | path

Searched defs:new_cast_node (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dmerge_cast_opt.cc129 auto new_cast_node = each_graph->NewCNode(cast_inputs); in MergeCastOpt() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dreorder_ops.cc238 …auto new_cast_node = func_graph->NewCNode({NewValueNode(std::make_shared<Primitive>(prim::kPrimCas… in ReorderTypeInsensitiveCastDown() local
314 auto new_cast_node = in ReorderCastUpTypeInsensitive() local
Dconvert_bfloat16.cc283 auto new_cast_node = NewCastNode(func_graph, node, orig_output_type); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dscalar_op_pass.cc612 auto new_cast_node = in RunArithmeticCheckPass() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc482 auto new_cast_node = pass.NewCNode(new_cast_inputs, graph); in CreateCast() local