Home
last modified time | relevance | path

Searched defs:kernel_type (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_build_info.h51 KernelType kernel_type() const { return kernel_type_; } in kernel_type() function
Dkernel_query.cc99 KernelType kernel_type) { in KernelQuery()
Dkernel_build_info.cc146 void KernelBuildInfo::KernelBuildInfoBuilder::SetKernelType(const KernelType &kernel_type) { in SetKernelType()
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dcommon_subexpression_elimination_test.cc46 KernelType kernel_type) { in CreateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc48 KernelType kernel_type = AnfAlgo::GetKernelType(anf_node); in SerialCompileImpl() local
83 KernelType kernel_type = AnfAlgo::GetKernelType(anf_node); in KernelBuildParallelCompile() local
Dkernel_select_ascend.cc495 KernelSelectStatus SelectKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SelectKernelInfo()
544 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SetKernelInfo()
/third_party/boost/boost/gil/extension/numeric/
Dalgorithm.hpp200 using kernel_type = typename std::iterator_traits<KernelIterator>::value_type; in correlate_pixels_k() typedef
/third_party/mindspore/mindspore/ccsrc/profiler/device/gpu/
Dgpu_profiling.h72 std::string kernel_type; member
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc398 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SetKernelInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_scheduler.cc930 auto kernel_type = KernelTransformType::kUnknown; in LinkDataArrow() local
976 KernelTransformType kernel_type; in LinkDataArrowForInternalParameter() local
1527 auto kernel_type = KernelTransformType::kUnknown; in LinkOutputResultArrowForOutputActor() local
2296 KernelTransformType *const kernel_type, in FetchKernelTransformTypeAndName()
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_session.cc346 auto kernel_type = kernel->desc().data_type; in MixPrecisionPreProcess() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dascend_kernel_compile.cc526 KernelType kernel_type = AnfAlgo::GetKernelType(anf_node); in GetAllAscendNodes() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_helper.cc604 void ResetKernelInfo(const AnfNodePtr &node, KernelType kernel_type) { in ResetKernelInfo()