| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | boundingbox_encode_cpu_kernel.cc | 31 if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("means")->isa<ValueTuple>() || in InitKernel()
32 AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("means")->isa<ValueList>()) { in InitKernel()
34 } else if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("means")->isa<FloatImm>()) { in InitKernel()
43 if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("stds")->isa<ValueTuple>() || in InitKernel()
44 AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("stds")->isa<ValueList>()) { in InitKernel()
46 } else if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("stds")->isa<FloatImm>()) { in InitKernel()
|
| D | boundingbox_decode_cpu_kernel.cc | 31 if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("means")->isa<ValueTuple>() || in InitKernel()
32 AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("means")->isa<ValueList>()) { in InitKernel()
34 } else if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("means")->isa<FloatImm>()) { in InitKernel()
43 if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("stds")->isa<ValueTuple>() || in InitKernel()
44 AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("stds")->isa<ValueList>()) { in InitKernel()
46 } else if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("stds")->isa<FloatImm>()) { in InitKernel()
|
| D | multinomial_cpu_kernel.cc | 33 …seed_ = static_cast<int>(GetValue<int64_t>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("seed"… in InitKernel()
34 …seed2_ = static_cast<int>(GetValue<int64_t>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("seed… in InitKernel()
|
| D | reduce_scatter_cpu_kernel.cc | 35 auto primitive = AnfAlgo::GetCNodePrimitive(kernel_node); in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_util.cc | 149 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomCount()
190 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomOperationType()
215 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomRootId()
229 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomSrcRank()
243 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomDestRank()
257 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomReceiveType()
269 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomGroup()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/ |
| D | fake_learned_scale_quant_perlayer_gpu_kernel.cc | 53 …quant_delay_ = static_cast<int>(GetValue<int64_t>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr… in Init()
54 training_ = GetValue<bool>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("training")); in Init()
55 neg_trunc_ = GetValue<bool>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("neg_trunc")); in Init()
|
| D | fake_learned_scale_quant_perchannel_gpu_kernel.cc | 61 …quant_delay_ = static_cast<int>(GetValue<int64_t>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr… in Init()
62 training_ = GetValue<bool>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("training")); in Init()
63 neg_trunc_ = GetValue<bool>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("neg_trunc")); in Init()
|
| D | fake_learned_scale_quant_perlayer_grad_gpu_kernel.cc | 50 …quant_delay_ = static_cast<int>(GetValue<int64_t>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr… in Init()
55 neg_trunc_ = GetValue<bool>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("neg_trunc")); in Init()
|
| D | fake_learned_scale_quant_perchannel_grad_gpu_kernel.cc | 56 …quant_delay_ = static_cast<int>(GetValue<int64_t>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr… in Init()
61 neg_trunc_ = GetValue<bool>(AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("neg_trunc")); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_cse.cc | 31 auto main_primitive = AnfAlgo::GetCNodePrimitive(main); in IsCNodePrimitveEqual()
32 auto node_primitive = AnfAlgo::GetCNodePrimitive(node); in IsCNodePrimitveEqual()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | squeeze_fusion.cc | 86 auto unsqueeze_primitive = GetCNodePrimitive(unsqueeze_node); in Process()
87 auto squeeze_primitive = GetCNodePrimitive(squeeze_node); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | batch_norm_relu_fusion.cc | 46 auto is_train = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("is_training"); in Process()
51 auto format_attr = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("format"); in Process()
|
| D | batch_norm_add_relu_fusion.cc | 49 auto is_train = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("is_training"); in Process()
54 auto format_attr = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("format"); in Process()
|
| D | post_batch_norm_add_relu_fusion.cc | 49 auto is_train = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("is_training"); in Process()
54 auto format_attr = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("format"); in Process()
|
| D | batch_norm_relu_grad_fusion.cc | 42 auto is_train = AnfAlgo::GetCNodePrimitive(node)->GetAttr("is_training"); in Process()
47 auto format_attr = AnfAlgo::GetCNodePrimitive(node)->GetAttr("format"); in Process()
|
| D | cudnn_inplace_fusion.cc | 59 auto primitive = AnfAlgo::GetCNodePrimitive(inplace_node); in SetPrimAttr()
125 auto first_node_prim = AnfAlgo::GetCNodePrimitive(first_node); in GetCoverIndex()
132 auto second_node_prim = AnfAlgo::GetCNodePrimitive(second_node); in GetCoverIndex()
|
| D | batch_norm_add_relu_grad_fusion.cc | 118 auto format_attr = AnfAlgo::GetCNodePrimitive(node)->GetAttr("format"); in PatternCheck()
187 auto is_train = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("is_training"); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | cse.cc | 124 auto prim = GetCNodePrimitive(node); in HasSideEffect()
137 auto prim_main = GetCNodePrimitive(main); in CheckRandomEffect()
138 auto prim_node = GetCNodePrimitive(node); in CheckRandomEffect()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 122 auto prim = GetCNodePrimitive(cnode); in InsertVirtualAssignAdd()
245 auto prim1 = GetCNodePrimitive(cnode1); in CompFunc()
246 auto prim2 = GetCNodePrimitive(cnode2); in CompFunc()
495 auto prim = GetCNodePrimitive(node); in LastStageEndNode()
501 auto temp_prim = GetCNodePrimitive(temp_node); in LastStageEndNode()
509 auto end_prim = GetCNodePrimitive(end_node); in LastStageEndNode()
547 auto prim = GetCNodePrimitive(node); in ParameterStartNode()
593 auto prim = GetCNodePrimitive(node); in GetBorderNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | graph_partitioner.cc | 41 const std::string &op_name = AnfAlgo::GetCNodePrimitive(node)->name(); in WeightCheck()
79 const std::string &op_name = AnfAlgo::GetCNodePrimitive(node)->name(); in CollectNodeInfo()
242 …if (graph_id.find("T_") != graph_id.npos && AnfAlgo::GetCNodePrimitive(item.first)->name() != kRet… in CollectSegments()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | send.cc | 33 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in Init()
|
| D | label_goto.cc | 38 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in Init()
|
| D | label_set.cc | 38 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in Init()
|
| D | recv.cc | 34 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/ |
| D | sync_batch_norm_grad_gpu_kernel.h | 68 auto root_rank = AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr(kAttrRootRank); in Init()
119 auto comm_stream_attr = AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("stream_id"); in Init()
|