Home
last modified time | relevance | path

Searched defs:kernel_node (Results 1 – 25 of 468) sorted by relevance

12345678910>>...19

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_query.cc31 void FilterInvalidKernelInfo(const CNodePtr &kernel_node, in FilterInvalidKernelInfo()
79 void KernelQueryAll(const CNodePtr &kernel_node, in KernelQueryAll()
98 void KernelQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel::KernelBuildInfo>>… in KernelQuery()
128 void AICPUQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel::KernelBuildInfo>> … in AICPUQuery()
136 bool IsSupportedByAICPU(const AnfNodePtr &kernel_node, const KernelBuildInfoPtr &select_kernel_buil… in IsSupportedByAICPU()
150 bool IsSupportedByAICore(const AnfNodePtr &kernel_node, const KernelBuildInfoPtr &select_kernel_bui… in IsSupportedByAICore()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dkernel_select_cpu.cc36 bool IsInputNotCNode(const CNodePtr &kernel_node, size_t input_index) { in IsInputNotCNode()
46 const CNodePtr &kernel_node) { in UpdatePrevNotCNodeFormatDtype()
63 void GetOutputDtypes(const CNodePtr &kernel_node, std::vector<TypeId> *output_types) { in GetOutputDtypes()
72 void GetOutputFormat(const CNodePtr &kernel_node, std::vector<std::string> *output_formats) { in GetOutputFormat()
79 void GetInputDtypes(const CNodePtr &kernel_node, std::vector<TypeId> *input_types, in GetInputDtypes()
94 void GetInputFormat(const CNodePtr &kernel_node, std::vector<std::string> *input_formats) { in GetInputFormat()
101 void GetOutputFormatsAndDtypes(const CNodePtr &kernel_node, const KernelAttr &kernel_attr, in GetOutputFormatsAndDtypes()
163 void ExpandKernelAttr(const CNodePtr &kernel_node, KernelAttr *kernel_attr) { in ExpandKernelAttr()
180 AnfNode *kernel_node) { in SetKernelBuildInfo()
190 void KernelNotSupportException(const AnfNodePtr &kernel_node, const std::vector<TypeId> &input_type… in KernelNotSupportException()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc73 std::string SupportedTypeList(const CNodePtr &kernel_node) { in SupportedTypeList()
103 bool SelectAkgKernel(const CNodePtr &kernel_node, const std::shared_ptr<KernelBuildInfo> &selected_… in SelectAkgKernel()
144 …ensorDeviceInfo(const kernel::KernelBuildInfo &selected_kernel_info, const CNodePtr &kernel_node) { in SetTensorDeviceInfo()
202 bool IsNeedProcessFormatInfo(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type) { in IsNeedProcessFormatInfo()
247 void UpdateKernelFormatInfo(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type, in UpdateKernelFormatInfo()
288 void SetGraphKernelInfo(const CNodePtr &kernel_node, const FuncGraphPtr &func_graph) { in SetGraphKernelInfo()
344 void PrintUnsupportedTypeException(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_t… in PrintUnsupportedTypeException()
398 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SetKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dsplit_cpu_kernel.cc29 void SplitCPUKernel<T>::InitKernel(const CNodePtr &kernel_node) { in InitKernel()
48 void SplitCPUKernel<T>::InitInputOutputSize(const CNodePtr &kernel_node) { in InitInputOutputSize()
105 void SplitCPUKernel<T>::CheckParam(const CNodePtr &kernel_node) { in CheckParam()
Dunique_cpu_kernel.cc23 void UniqueCPUKernel::InitKernel(const CNodePtr &kernel_node) { in InitKernel()
37 void UniqueCPUKernel::InitInputOutputSize(const CNodePtr &kernel_node) { in InitInputOutputSize()
Dunpack_cpu_kernel.cc29 void UnpackCPUKernel<T>::InitKernel(const CNodePtr &kernel_node) { in InitKernel()
56 void UnpackCPUKernel<T>::InitInputOutputSize(const CNodePtr &kernel_node) { in InitInputOutputSize()
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/
Dkernel_factory.h61 K Create(const std::string &name, const CNodePtr &kernel_node) { in Create()
82 virtual bool Matched(const ParamsInfo &params_info, const CNodePtr &kernel_node) { return true; } in Matched()
Doptimizer_kernel.h45 virtual void InitKernel(const CNodePtr &kernel_node) {} in InitKernel()
67 void InitServerKernelInputOutputSize(const CNodePtr &kernel_node) { in InitServerKernelInputOutputSize()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_ascend.cc136 …Counts(const kernel::KernelBuildInfo &kernel_build_info, const std::shared_ptr<CNode> &kernel_node, in UpdateCurMatchCounts()
204 …const CNodePtr &kernel_node, const std::vector<std::shared_ptr<kernel::KernelBuildInfo>> &kernel_i… in ChooseMatchedKernelInfo()
328 void SetCastAndWeightFormat(const CNodePtr &kernel_node) { in SetCastAndWeightFormat()
360 …(const AnfNodePtr &real_input_node, std::vector<string> output_format, const CNodePtr &kernel_node, in SetWeightFormat()
432 void SetTensorDeviceInfo(const CNodePtr &kernel_node) { in SetTensorDeviceInfo()
456 KernelSelectStatus SetMatchedKernelInfo(const CNodePtr &kernel_node, in SetMatchedKernelInfo()
495 KernelSelectStatus SelectKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SelectKernelInfo()
544 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SetKernelInfo()
Dkernel_select_graph_kernel.cc61 void ResetKernelBuildInfo(const CNodePtr &kernel_node) { in ResetKernelBuildInfo()
195 void GetDefaultFormat(const CNodePtr &kernel_node, std::string *default_format, bool *use_same_form… in GetDefaultFormat()
245 void UpdateInputsKernelInfo(const CNodePtr &kernel_node, const std::vector<AnfNodePtr> &input_list, in UpdateInputsKernelInfo()
343 void CheckFormatsAndDtypes(const CNodePtr &kernel_node, const std::vector<AnfNodePtr> &input_list, in CheckFormatsAndDtypes()
395 void UpdateFormatsAndDtypes(const CNodePtr &kernel_node, const std::vector<AnfNodePtr> &node_list, in UpdateFormatsAndDtypes()
448 void SetGraphKernelInfo(const CNodePtr &kernel_node, const std::vector<std::pair<AnfNodePtr, size_t… in SetGraphKernelInfo()
481 void SelectGraphKernelInfo(const CNodePtr &kernel_node, const FuncGraphPtr &func_graph) { in SelectGraphKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_metadata.cc27 void AicpuMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *… in AicpuMetadataInfo()
51 void AicpuMetadataInfoForSpecialNodes(const CNodePtr &kernel_node, in AicpuMetadataInfoForSpecialNodes()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dpooling_gpu_kernel.h78 bool Init(const CNodePtr &kernel_node) { in Init()
163 bool CheckParam(const CNodePtr &kernel_node) { in CheckParam()
172 void SetPoolingMode(const CNodePtr &kernel_node) { in SetPoolingMode()
182 void SetPad(const CNodePtr &kernel_node) { in SetPad()
225 void SetPad3D(const CNodePtr &kernel_node) { in SetPad3D()
Dpooling_grad_gpu_kernel.h85 …bool InitShape(const CNodePtr &kernel_node, int *dimA, int *strideAin, int *dimAy, int *strideAiny… in InitShape()
121 bool Init(const CNodePtr &kernel_node) override { in Init()
213 bool CheckParam(const CNodePtr &kernel_node) { in CheckParam()
222 void SetPad(const CNodePtr &kernel_node) { in SetPad()
271 void SetPad3D(const CNodePtr &kernel_node) { in SetPad3D()
329 void SetPoolingMode(const CNodePtr &kernel_node) { in SetPoolingMode()
Dim2col_gpu_kernel.h84 bool Init(const CNodePtr &kernel_node) override { in Init()
191 bool CheckParam(const CNodePtr &kernel_node) { in CheckParam()
206 void SetPad(const std::vector<size_t> &in_shape, const CNodePtr &kernel_node) { in SetPad()
262 void SetStrideAndDilation(const CNodePtr &kernel_node) { in SetStrideAndDilation()
Dconv3d_grad_input_gpu_kernel.h83 bool Init(const CNodePtr &kernel_node) override { in Init()
279 bool CheckParam(const CNodePtr &kernel_node) { in CheckParam()
293 void SetPad(const std::vector<int> &input_shape, const CNodePtr &kernel_node) { in SetPad()
312 void GetInputShape(const CNodePtr &kernel_node, std::vector<size_t> *input_shape) { in GetInputShape()
345 void SetStrideAndDilation(const CNodePtr &kernel_node) { in SetStrideAndDilation()
Dconv2d_grad_input_gpu_kernel.h116 bool Init(const CNodePtr &kernel_node) override { in Init()
282 bool CheckParam(const CNodePtr &kernel_node) { in CheckParam()
295 void SetPad(const std::vector<int> &input_shape, const CNodePtr &kernel_node) { in SetPad()
324 void GetInputShape(const CNodePtr &kernel_node, std::vector<size_t> *input_shape) { in GetInputShape()
355 void SetStrideAndDilation(const CNodePtr &kernel_node) { in SetStrideAndDilation()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dstrided_slice_gpu_common.h33 void CollectInfo(const CNodePtr &kernel_node) { in CollectInfo()
41 void FillEmptyDims(const CNodePtr &kernel_node) { in FillEmptyDims()
71 void ParseMasks(const CNodePtr &kernel_node) { in ParseMasks()
Dtensor_copy_slices_gpu_kernel.h59 bool Init(const CNodePtr &kernel_node) override { in Init()
110 void CheckAtrrAndShapeValid(const CNodePtr &kernel_node) { in CheckAtrrAndShapeValid()
150 void FillEmptyDims(const CNodePtr &kernel_node) { in FillEmptyDims()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/
Dpush_kernel.h56 void Init(const CNodePtr &kernel_node) { in Init()
79 void InitKernel(const CNodePtr &kernel_node) { return; } in InitKernel()
Dpull_kernel.h46 void Init(const CNodePtr &kernel_node) { in Init()
72 void InitKernel(const CNodePtr &kernel_node) { return; } in InitKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/
Dpush_metrics_kernel.h98 void Init(const CNodePtr &kernel_node) { in Init()
107 void InitKernel(const CNodePtr &kernel_node) { return; } in InitKernel()
Dstart_fl_job_kernel.h72 void Init(const CNodePtr &kernel_node) { in Init()
95 void InitKernel(const CNodePtr &kernel_node) { return; } in InitKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/rl/
Dbuffer_get_cpu_kernel.h33 void Init(const CNodePtr &kernel_node) { in Init()
84 void InitKernel(const CNodePtr &kernel_node) { return; } in InitKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/other/
Dassign_gpu_kernel.h54 bool Init(const CNodePtr &kernel_node) override { in Init()
83 bool CheckParam(const CNodePtr &kernel_node) { in CheckParam()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.h59 virtual void Query(const CNodePtr &kernel_node, in Query()
81 virtual void GetTbeKernelMetaInfo(const CNodePtr &kernel_node, in GetTbeKernelMetaInfo()

12345678910>>...19