| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/ |
| D | convert_cast_format.cc | 55 auto cast_node = input_node->cast<CNodePtr>(); in Process() local
62 void ConvertCastFormat::SetCastFormat(const CNodePtr &cast_node, const string &format) const { in SetCastFormat() argument
72 void ConvertCastFormat::ChangeCastFormat(const CNodePtr &cast_node, const FuncGraphPtr &func_graph)… in ChangeCastFormat() argument
|
| D | merge_cast_to_op.cc | 153 static bool IsCastFp16ToFp32(const CNodePtr &cast_node) { in IsCastFp16ToFp32()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/ |
| D | cast_matmul_fusion.cc | 28 void UpdateBuildInfo(const AnfNodePtr &matmul_node, const AnfNodePtr &cast_node) { in UpdateBuildInfo()
67 auto cast_node = cnode->inputs().back(); // bias node in DoFuse() local
|
| D | reorder_ops.cc | 278 auto cast_node = node_input->cast<CNodePtr>(); in ReorderCastUpTypeInsensitive() local
295 …if (!std::all_of(cast_nodes.begin(), cast_nodes.end(), [&small_type](const AnfNodePtr &cast_node) { in ReorderCastUpTypeInsensitive()
|
| D | convert_bfloat16.cc | 88 auto cast_node = in NewCastNode() local
186 auto cast_node = input_node->cast<CNodePtr>(); in CastInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/ |
| D | merge_cast_opt.cc | 95 for (const auto &cast_node : pair.second) { in MergeCastOpt() local
131 for (const auto &cast_node : pair.second) { in MergeCastOpt() local
|
| D | remove_cast_before_assign_add.cc | 64 auto cast_node = assign_add_node->input(kIndex2)->cast<CNodePtr>(); in RemoveCastBeforeAssignAdd() local
|
| D | float32_redistribution.cc | 41 auto cast_node = graph->NewCNode({NewValueNode(prim), pre_node, type_node}); in InsertCastCNode() local
|
| D | assign_add_opt.cc | 201 CNodePtr cast_node = nullptr; in MergeMultiMatmulAssignAdd() local
295 auto cast_node = assign_add_node->cast<CNodePtr>()->input(kIndex2)->cast<CNodePtr>(); in SkipAssignAddEliminate() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/ |
| D | insert_cast_to_pyexecute.cc | 54 …auto cast_node = insert_cast_function_(fg, node, kOpFormat_DEFAULT, default_type, default_type, no… in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/ |
| D | bce_with_logits_loss_for_ge.cc | 60 auto cast_node = AddCastNode(graph, target_type_id, sigmoid_node, false); in Process() local
|
| D | fused_cast_add.cc | 49 const auto &cast_node = m.Get(kCast16); in SelectAddInput() local
|
| D | add_cast_for_ge.cc | 92 auto cast_node = AddCastNode(graph, cast_info.dst_dtype, node, true, input_idx); in AddCastForGeInput() local
|
| D | dropout_for_ge.cc | 131 auto cast_node = node->func_graph()->NewCNode( in GenCastNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | bce_with_logits_loss_fission.cc | 91 auto cast_node = NewCNode(cast_inputs, func_graph); in AddReduceNode() local
|
| D | dynamic_gru_v2_grad_fission.cc | 167 auto cast_node = CastNodeToSpecificDType(func_graph, dgate_h, kNumberTypeFloat16); in AddTLoopNode() local
332 auto cast_node = CastNodeToSpecificDType(func_graph, node, kNumberTypeFloat16); in AddDwhMatmulNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | apply_momentum_weight_scale_fusion.cc | 84 auto cast_node = node->cast<CNodePtr>()->input(kInputIndex); in GetCastInput() local
|
| D | combine_cast_fusion.cc | 41 for (auto &cast_node : node_list) { in GetDealList() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | add_cast_rms_norm_cast_fusion.cc | 107 auto cast_node = common::AnfAlgo::GetInputNode(utils::cast<CNodePtr>(rms_norm_node), 0); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | kvcache_quant_pass.cc | 231 auto cast_node = add_node->cast<CNodePtr>()->input(kIndexOne); in RunAntiQuantPass() local
|
| D | scalar_op_pass.cc | 423 CNodePtr cast_node = nullptr; in ReplaceShapeTupleGet() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_inputs_adjust.cc | 508 auto cast_node = in AdjustROIAlign() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/ |
| D | kernel_select_ascend.cc | 149 auto cast_node = input_node->cast<CNodePtr>(); in RefreshCastAndParamWeightFormat() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ |
| D | flash_sp.cc | 286 auto cast_node = tensor_node->func_graph()->NewCNode(cast_inputs); in NewCastNode() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/optimizer/ |
| D | lib_test.cc | 175 auto cast_node = inputs[0]; in TEST_F() local
|