Home
last modified time | relevance | path

Searched refs:KernelInfo (Results 1 – 25 of 91) sorted by relevance

1234

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_info.cc21 const kernel::KernelBuildInfo *KernelInfo::select_kernel_build_info() const { return select_kernel_… in select_kernel_build_info()
23 kernel::KernelBuildInfoPtr KernelInfo::GetMutableSelectKernelBuildInfo() const { return select_kern… in GetMutableSelectKernelBuildInfo()
25 const DeviceAddress *KernelInfo::GetOutputAddr(size_t index) const { in GetOutputAddr()
33 DeviceAddressPtr KernelInfo::GetMutableOutputAddr(size_t index) const { in GetMutableOutputAddr()
41 bool KernelInfo::OutputAddrExist(size_t index) const { in OutputAddrExist()
48 bool KernelInfo::SetOutputAddr(const DeviceAddressPtr &output_address, size_t index) { in SetOutputAddr()
68 DeviceAddress *KernelInfo::GetWorkspaceAddr(size_t index) const { in GetWorkspaceAddr()
76 DeviceAddressPtr KernelInfo::GetMutableWorkspaceAddr(size_t index) const { in GetMutableWorkspaceAddr()
84 bool KernelInfo::WorkspaceAddrExist(size_t index) const { in WorkspaceAddrExist()
91 bool KernelInfo::SetWorkspaceAddr(const DeviceAddressPtr &output_address, size_t index) { in SetWorkspaceAddr()
[all …]
Dkernel_info.h31 class KernelInfo : public KernelInfoDevice {
33 KernelInfo() { in KernelInfo() function
43 virtual ~KernelInfo() = default;
71 bool operator==(const KernelInfo &other) const;
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc76 relu1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
77 relu2->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
91 cast->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
147 relu1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
148 relu2->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
149 relu3->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
150 relu4->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
151 relu5->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
152 relu6->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
170 cast->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
[all …]
/third_party/skia/third_party/externals/angle2/src/libANGLE/
DCLKernel.cpp23 cl_int Kernel::getInfo(KernelInfo name, size_t valueSize, void *value, size_t *valueSizeRet) const in getInfo()
32 case KernelInfo::FunctionName: in getInfo()
36 case KernelInfo::NumArgs: in getInfo()
40 case KernelInfo::ReferenceCount: in getInfo()
45 case KernelInfo::Context: in getInfo()
50 case KernelInfo::Program: in getInfo()
55 case KernelInfo::Attributes: in getInfo()
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc28 using device::KernelInfo;
252 add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
253 auto d_kernel_info = dynamic_cast<KernelInfo *>(add->kernel_info()); in TEST_F()
271 add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
272 auto d_kernel_info = dynamic_cast<KernelInfo *>(add->kernel_info()); in TEST_F()
290 pre_add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
291 auto d_kernel_info = dynamic_cast<KernelInfo *>(pre_add->kernel_info()); in TEST_F()
370 add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
371 auto d_kernel_info = dynamic_cast<KernelInfo *>(add->kernel_info()); in TEST_F()
403 add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
[all …]
Dkernel_graph_test.cc27 using device::KernelInfo;
44 add_value->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
45 auto mutable_kernel_info = dynamic_cast<device::KernelInfo *>(add_value->kernel_info()); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dlayer_norm_beta_gamma_backprop_fusion_test.cc81 cast0->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
82 cast1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
100 layer->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
142 cast0->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
143 cast1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
159 layer->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
201 cast0->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
202 cast1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
218 layer->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
255 cast->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
[all …]
/third_party/skia/third_party/externals/angle2/src/common/
DPackedCLEnums_autogen.cpp1348 KernelInfo FromCLenum<KernelInfo>(CLenum from) in FromCLenum()
1353 return KernelInfo::FunctionName; in FromCLenum()
1355 return KernelInfo::NumArgs; in FromCLenum()
1357 return KernelInfo::ReferenceCount; in FromCLenum()
1359 return KernelInfo::Context; in FromCLenum()
1361 return KernelInfo::Program; in FromCLenum()
1363 return KernelInfo::Attributes; in FromCLenum()
1365 return KernelInfo::InvalidEnum; in FromCLenum()
1369 CLenum ToCLenum(KernelInfo from) in ToCLenum()
1373 case KernelInfo::FunctionName: in ToCLenum()
[all …]
DPackedCLEnums_autogen.h285 enum class KernelInfo : uint8_t enum
299 KernelInfo FromCLenum<KernelInfo>(CLenum from);
300 CLenum ToCLenum(KernelInfo from);
301 std::ostream &operator<<(std::ostream &os, KernelInfo value);
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/
Dinsert_cast_test.cc78 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
81 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
132 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
135 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
Dinsert_trans_op_test.cc57 add->set_kernel_info(std::make_shared<device::KernelInfo>()); in GetSingleOutputGraph()
81 max_pool->set_kernel_info(std::make_shared<device::KernelInfo>()); in GetMutilpleOutputGraph()
/third_party/skia/third_party/externals/angle2/src/libANGLE/renderer/cl/
DCLKernelCL.cpp59 bool GetKernelInfo(cl_kernel kernel, cl::KernelInfo name, T &value, cl_int &errorCode) in GetKernelInfo()
95 bool GetKernelString(cl_kernel kernel, cl::KernelInfo name, std::string &string, cl_int &errorCode) in GetKernelString()
175 if (!GetKernelString(mNative, cl::KernelInfo::FunctionName, info.functionName, errorCode) || in createInfo()
176 !GetKernelInfo(mNative, cl::KernelInfo::NumArgs, info.numArgs, errorCode) || in createInfo()
178 !GetKernelString(mNative, cl::KernelInfo::Attributes, info.attributes, errorCode))) in createInfo()
/third_party/skia/third_party/externals/swiftshader/third_party/llvm-10.0/llvm/lib/Target/AMDGPU/
DAMDGPUAsmPrinter.h69 void getAmdKernelCode(amd_kernel_code_t &Out, const SIProgramInfo &KernelInfo,
78 const SIProgramInfo &KernelInfo);
80 const SIProgramInfo &KernelInfo);
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dcommon_subexpression_elimination_test.cc102 add1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
103 add2->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
114 add1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
145 add1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
146 add2->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
Deliminate_redundant_op_test.cc109 sub->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
110 add->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
176 sub->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
177 add->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
253 sub->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
254 add->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
Dallreduce_fusion_test.cc66 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
107 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
149 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
205 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc71 relu1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
114 conv2d_backprop_filter->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
172 dynamic_rnn->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
226 layer_norm->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
271 relu1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
272 relu2->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
286 cast->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
343 add->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
345 conv2d->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc43 using device::KernelInfo;
665 auto kernel_info = dynamic_cast<device::KernelInfo *>(node->kernel_info()); in GetAllOutputFormats()
680 auto kernel_info = dynamic_cast<device::KernelInfo *>(node->kernel_info()); in GetAllInputFormats()
695 auto kernel_info = dynamic_cast<device::KernelInfo *>(node->kernel_info()); in GetAllInputDeviceTypes()
710 auto kernel_info = dynamic_cast<device::KernelInfo *>(node->kernel_info()); in GetAllOutputDeviceTypes()
725 auto kernel_info = dynamic_cast<device::KernelInfo *>(node->kernel_info()); in GetOriginDataFormat()
744 auto kernel_info = dynamic_cast<device::KernelInfo *>(node->kernel_info()); in GetOutputFormat()
768 auto kernel_info = dynamic_cast<device::KernelInfo *>(node->kernel_info()); in GetInputFormat()
913 auto kernel_info = dynamic_cast<device::KernelInfo *>(node->kernel_info()); in GetInputReshapeType()
933 auto kernel_info = dynamic_cast<device::KernelInfo *>(node->kernel_info()); in GetOutputReshapeType()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dkernel_actor.h36 using mindspore::device::KernelInfo;
100 KernelInfo *kernel_info_;
Ddata_source_actor.h36 using mindspore::device::KernelInfo;
113 KernelInfo *kernel_info_{nullptr};
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_cse.cc81 auto main_kernel_info = dynamic_cast<device::KernelInfo *>(main->kernel_info()); in CheckEqualKernelBuildInfo()
82 auto node_kernel_info = dynamic_cast<device::KernelInfo *>(node->kernel_info()); in CheckEqualKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dcommon_subexpression_elimination.cc66 auto main_kernel_info = dynamic_cast<device::KernelInfo *>(main->kernel_info()); in CheckEqualKernelBuildInfo()
67 auto node_kernel_info = dynamic_cast<device::KernelInfo *>(node->kernel_info()); in CheckEqualKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dbatch_norm_grad_split.cc42 bn_update_grad->set_kernel_info(std::make_shared<device::KernelInfo>()); in CreateOutputsOfUpdateGrad()
76 bn_reduce_grad->set_kernel_info(std::make_shared<device::KernelInfo>()); in CreateOutputsOfReduceGrad()
Dbn_grad_split.cc43 bn_update_grad->set_kernel_info(std::make_shared<device::KernelInfo>()); in CreateOutputsOfUpdateGrad()
75 bn_reduce_grad->set_kernel_info(std::make_shared<device::KernelInfo>()); in CreateOutputsOfReduceGrad()
/third_party/mindspore/mindspore/ccsrc/profiler/device/gpu/
Dgpu_profiling.h58 struct KernelInfo { struct
86 KernelInfo kernel_info; argument

1234