| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/ |
| D | kernel_build_utils.cc | 44 bool IsInputNotCNode(const CNodePtr &kernel_node, size_t input_index) { in IsInputNotCNode()
53 void GetOutputDtypes(const CNodePtr &kernel_node, std::vector<TypeId> *output_types) { in GetOutputDtypes()
61 void GetOutputFormat(const CNodePtr &kernel_node, std::vector<std::string> *output_formats) { in GetOutputFormat()
68 void GetInputDtypes(const CNodePtr &kernel_node, std::vector<TypeId> *input_types, in GetInputDtypes()
83 void GetInputFormat(const CNodePtr &kernel_node, std::vector<std::string> *input_formats) { in GetInputFormat()
142 void ExpandKernelAttr(const CNodePtr &kernel_node, kernel::KernelAttr *kernel_attr) { in ExpandKernelAttr()
172 AnfNode *kernel_node) { in SetKernelBuildInfo()
182 void UpdateDynamicKernelBuildInfo(const CNodePtr &kernel_node) { in UpdateDynamicKernelBuildInfo()
198 void UpdateCustomKernelBuildInfo(const CNodePtr &kernel_node, bool is_akg_op) { in UpdateCustomKernelBuildInfo()
228 kernel::KernelAttr FillNoneInKernelAttr(const CNodePtr &kernel_node, const std::vector<TypeId> &inp… in FillNoneInKernelAttr()
[all …]
|
| D | runtime_utils.cc | 71 void RuntimeUtils::UpdateKernelNodeOutputInfo(const AnfNodePtr &kernel_node, in UpdateKernelNodeOutputInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | kernel_select_cpu.cc | 86 void GetOutputDtypes(const CNodePtr &kernel_node, std::vector<TypeId> *output_types) { in GetOutputDtypes()
95 void GetOutputDtypesForRealTuple(const CNodePtr &kernel_node, std::vector<TypeId> *output_types) { in GetOutputDtypesForRealTuple()
101 void GetOutputFormat(const CNodePtr &kernel_node, std::vector<std::string> *output_formats) { in GetOutputFormat()
108 void GetInputDtypes(const CNodePtr &kernel_node, std::vector<TypeId> *input_types) { in GetInputDtypes()
117 void GetInputFormat(const CNodePtr &kernel_node, std::vector<std::string> *input_formats) { in GetInputFormat()
178 void ExpandKernelAttr(const CNodePtr &kernel_node, kernel::KernelAttr *kernel_attr) { in ExpandKernelAttr()
224 void ExpandMultiDynamicAttr(const CNodePtr &kernel_node, const std::vector<int64_t> &dyn_input_size… in ExpandMultiDynamicAttr()
277 void ExpandKernelAttrByDynamicSize(const CNodePtr &kernel_node, kernel::KernelAttr *kernel_attr, bo… in ExpandKernelAttrByDynamicSize()
292 AnfNode *kernel_node) { in SetKernelBuildInfo()
304 void SetKernelBuildInfoWithSelectedAttr(const CNodePtr &kernel_node, const kernel::KernelAttr &sele… in SetKernelBuildInfoWithSelectedAttr()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | kernel_info_setter.cc | 120 std::string GetSupportedTypesStr(const CNodePtr &kernel_node, KernelType kernel_type) { in GetSupportedTypesStr()
181 bool SelectAkgKernel(const CNodePtr &kernel_node, const std::shared_ptr<KernelBuildInfo> &selected_… in SelectAkgKernel()
216 bool SelectCustomKernel(const CNodePtr &kernel_node, const std::shared_ptr<KernelBuildInfo> &select… in SelectCustomKernel()
262 …tTensorDeviceInfo(const kernel::KernelBuildInfo &selected_kernel_info, const CNodePtr &kernel_node, in SetTensorDeviceInfo()
329 bool IsNeedProcessFormatInfo(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type) { in IsNeedProcessFormatInfo()
370 void UpdateKernelFormatInfo(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type, in UpdateKernelFormatInfo()
411 void SetGraphKernelInfo(const CNodePtr &kernel_node, const FuncGraphPtr &func_graph) { in SetGraphKernelInfo()
473 std::pair<std::string, ExceptionType> PrintUnsupportedTypeWarning(const CNodePtr &kernel_node, in PrintUnsupportedTypeWarning()
557 bool GetSelectKernelResult(const CNodePtr &kernel_node, in GetSelectKernelResult()
610 …std::pair<std::string, ExceptionType>> GetSelectKernelObjectTypeResult(const CNodePtr &kernel_node, in GetSelectKernelObjectTypeResult()
[all …]
|
| D | gpu_stream_assign.cc | 49 for (const auto &kernel_node : execution_kernels) { in AssignGpuStream() local
84 for (const auto &kernel_node : kernel_graph->execution_order()) { in AssignDefaultGpuStream() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/ |
| D | aicpu_attr_and_input_convert_regist.cc | 68 void ConvertAttrToInput(const CNodePtr &kernel_node, std::vector<std::pair<string, size_t>> *infos)… in ConvertAttrToInput()
182 bool GetAicpuOpAttrToInputInfo(const CNodePtr &kernel_node, std::vector<std::pair<string, size_t>> … in GetAicpuOpAttrToInputInfo()
193 bool GetAicpuOpInputToAttrInfo(const CNodePtr &kernel_node, std::map<size_t, std::string> *input_to… in GetAicpuOpInputToAttrInfo()
204 void ConvertAttrAndInputBeforeAicpuKernelSelect(const CNodePtr &kernel_node) { in ConvertAttrAndInputBeforeAicpuKernelSelect()
|
| D | aicpu_kernel_metadata.cc | 30 void AicpuMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *… in AicpuMetadataInfo()
59 void AicpuMetadataInfoForSpecialNodes(const CNodePtr &kernel_node, in AicpuMetadataInfoForSpecialNodes()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/ |
| D | hccl_kernel_metadata.cc | 37 std::string GetKernelFormat(const CNodePtr &kernel_node, size_t index) { in GetKernelFormat()
74 void HcclMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *k… in HcclMetadataInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/bisheng/ |
| D | bisheng_kernel_metadata.cc | 24 void BiShengMetadataInfo(const CNodePtr &kernel_node, std::vector<KernelBuildInfoPtr> *const kernel… in BiShengMetadataInfo()
|
| D | custom_bisheng_kernel.cc | 38 bool CustomBiShengKernel::InitKernel(const AnfNodePtr &kernel_node) { in InitKernel()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/stub/dynamic_shape/ |
| D | dynamic_shape_stub.cc | 22 void HostMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *k… in HostMetadataInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/akg/ |
| D | akg_kernel_metadata.cc | 25 void AkgMetadataInfo(const CNodePtr &kernel_node, std::vector<KernelBuildInfoPtr> *kernel_info_list… in AkgMetadataInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/ |
| D | kernel_select_ascend.cc | 83 …(const AnfNodePtr &real_input_node, std::vector<string> output_format, const CNodePtr &kernel_node, in SetWeightFormat()
171 void RefreshInputParameter(const CNodePtr &kernel_node, const AnfNodePtr &input_kernel_node, in RefreshInputParameter()
188 void SetTensorDeviceInfo(const CNodePtr &kernel_node) { in SetTensorDeviceInfo()
329 bool SetMatchKernelInfo(const CNodePtr &kernel_node, const std::vector<kernel::KernelBuildInfoPtr> … in SetMatchKernelInfo()
413 TypeId GetInputDeviceType(const AnfNodePtr &kernel_node, size_t input_idx) { in GetInputDeviceType()
788 void SetKernelInfo(const CNodePtr &kernel_node, KernelType) override { in SetKernelInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/ |
| D | cpu_session.cc | 251 for (const auto &kernel_node : kernel_nodes) { in SetKernelInfo() local
258 void KernelNotSupportException(const AnfNodePtr &kernel_node) { in KernelNotSupportException()
306 for (const auto &kernel_node : kernel_nodes) { in BuildKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/host/ |
| D | host_kernel_metadata.cc | 26 void HostMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *k… in HostMetadataInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | common_utils.cc | 277 std::vector<KernelObjectType> CalInputKernelObjectTypes(const AnfNodePtr &kernel_node, in CalInputKernelObjectTypes()
286 std::vector<KernelObjectType> CalOutputKernelObjectTypes(const AnfNodePtr &kernel_node, in CalOutputKernelObjectTypes()
295 std::vector<KernelObjectType> CalOutputElementObjectTypes(const AnfNodePtr &kernel_node, in CalOutputElementObjectTypes()
327 void SetKernelObjectTypeBuildInfo(const AnfNodePtr &kernel_node, in SetKernelObjectTypeBuildInfo()
345 void SetKernelObjectTypeBuildInfo(const AnfNodePtr &kernel_node, in SetKernelObjectTypeBuildInfo()
371 void SetKernelObjectTypeWithSelectedAttr(const CNodePtr &kernel_node, const kernel::KernelAttr &sel… in SetKernelObjectTypeWithSelectedAttr()
736 KernelAttr GetKernelAttrFromNode(const AnfNodePtr &kernel_node) { in GetKernelAttrFromNode()
|
| D | framework_utils.cc | 445 const CNodePtr &kernel_node) { in SetKernelBuildInfo()
491 bool ParseMetadata(const CNodePtr &kernel_node, const std::shared_ptr<const OpInfo> &op_info_ptr, P… in ParseMetadata()
805 bool SelectKernelByObjectType(const CNodePtr &kernel_node, const std::vector<KernelAttr> ®istere… in SelectKernelByObjectType()
839 std::pair<std::string, ExceptionType> KernelObjectTypeNotSupportWarning(const CNodePtr &kernel_node… in KernelObjectTypeNotSupportWarning()
873 void UnfoldKernelBuildInfo(const CNodePtr &kernel_node) { in UnfoldKernelBuildInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/rts/ |
| D | rt_kernel_info.cc | 56 void GetRtKelInfo(const CNodePtr &kernel_node, in GetRtKelInfo()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/pre_activate/ascend/format_type/ |
| D | merge_cast_to_op_test.cc | 48 void Query(const CNodePtr &kernel_node, in Query()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/ |
| D | insert_cast_cpu.cc | 150 auto kernel_node = common::AnfAlgo::VisitKernel(input_node, 0).first; in InsertCastForGraphOutput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | multi_margin_loss_cpu_kernel.cc | 220 void MultiMarginLossCPUKernelMod::CheckParam(const CNodePtr &kernel_node) { in CheckParam()
|
| D | crop_and_resize_cpu_kernel.cc | 373 void CropAndResizeCpuKernelMod::InitFunc(const CNodePtr &kernel_node) { in InitFunc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_pad_fusion.cc | 95 auto kernel_node = conv_primitive->GetAttr(ops::kKernelSize); in ReplaceParamsAndNodes() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/ |
| D | embedding_cache_scheduler.cc | 376 for (const auto &kernel_node : graph->execution_order()) { in SetDataSetChannel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/ |
| D | gpu_session.cc | 131 for (const auto &kernel_node : kernel_graph->execution_order()) { in SelectKernel() local
|