| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_ascend.h | 31 KernelType kernel_type = KernelType::UNKNOWN_KERNEL_TYPE);
34 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type);
|
| D | kernel_select_ascend.cc | 495 KernelSelectStatus SelectKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SelectKernelInfo() argument
505 kernel::KernelQuery(kernel_node, &kernel_info_list, kernel_type); in SelectKernelInfo()
544 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SetKernelInfo() argument
565 auto new_kernel_type = kernel_type; in SetKernelInfo()
567 if (kernel_type == UNKNOWN_KERNEL_TYPE) { in SetKernelInfo()
570 kernel::KernelQuery(kernel_node, &kernel_info_list, kernel_type); in SetKernelInfo()
|
| D | kernel_build_ascend.cc | 48 KernelType kernel_type = AnfAlgo::GetKernelType(anf_node); in SerialCompileImpl() local
49 switch (kernel_type) { in SerialCompileImpl()
68 …G(EXCEPTION) << "node [" << anf_node->DebugString() << "] Unsupported kernel_type:" << kernel_type; in SerialCompileImpl()
83 KernelType kernel_type = AnfAlgo::GetKernelType(anf_node); in KernelBuildParallelCompile() local
84 switch (kernel_type) { in KernelBuildParallelCompile()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_query.cc | 99 KernelType kernel_type) { in KernelQuery() argument
108 kernel_type = KernelType::AKG_KERNEL; in KernelQuery()
113 kernel_type = KernelType::AKG_KERNEL; in KernelQuery()
116 switch (kernel_type) { in KernelQuery()
|
| D | kernel_build_info.h | 51 KernelType kernel_type() const { return kernel_type_; } in kernel_type() function
130 SetKernelType(kernel_build_info->kernel_type()); in KernelBuildInfoBuilder()
149 void SetKernelType(const KernelType &kernel_type);
|
| D | kernel_query.h | 29 KernelType kernel_type = KernelType::UNKNOWN_KERNEL_TYPE);
|
| D | kernel_build_info.cc | 146 void KernelBuildInfo::KernelBuildInfoBuilder::SetKernelType(const KernelType &kernel_type) { in SetKernelType() argument
148 kernel_build_info_->kernel_type_ = kernel_type; in SetKernelType()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | dynamic_rnn_grad_reformat.cc | 55 matmul_new_builder->SetKernelType(matmul_kernel_build_info->kernel_type()); in Process()
69 split_new_builder->SetKernelType(split_kernel_build_info->kernel_type()); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.cc | 398 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SetKernelInfo() argument
431 if (kernel_type == UNKNOWN_KERNEL_TYPE) { in SetKernelInfo()
439 kernel_type = AKG_KERNEL; in SetKernelInfo()
441 } else if (kernel_type == AKG_KERNEL) { in SetKernelInfo()
448 builder->SetKernelType(kernel_type); in SetKernelInfo()
|
| D | kernel_info_setter.h | 70 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type = KernelType::UNKNOWN_KERNEL…
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_scheduler.cc | 930 auto kernel_type = KernelTransformType::kUnknown; in LinkDataArrow() local
932 …FetchKernelTransformTypeAndName(from_kernel, graph, graph_compiler_info, &kernel_type, &kernel_nam… in LinkDataArrow()
934 if (kKernelTypeToLinkFunc.count(kernel_type) > 0) { in LinkDataArrow()
935 (this->*kKernelTypeToLinkFunc[kernel_type])(from_actor, to_actor, from_kernel_with_output_idx, in LinkDataArrow()
976 KernelTransformType kernel_type; in LinkDataArrowForInternalParameter() local
978 kernel_type = KernelTransformType::kDeviceTensorStore; in LinkDataArrowForInternalParameter()
994 kernel_type = actor_pair.first->type_; in LinkDataArrowForInternalParameter()
997 if (kKernelTypeToLinkFunc.count(kernel_type) == 0) { in LinkDataArrowForInternalParameter()
998 … << "Invalid internal parameter:" << internal_parameter->DebugString() << ", type:" << kernel_type; in LinkDataArrowForInternalParameter()
1000 …(this->*kKernelTypeToLinkFunc[kernel_type])(real_from_actor, to_actor, real_from_kernel_with_outpu… in LinkDataArrowForInternalParameter()
[all …]
|
| D | graph_scheduler.h | 236 … KernelTransformType *const kernel_type, std::string *const kernel_name);
|
| /third_party/mindspore/mindspore/ccsrc/profiler/device/gpu/ |
| D | gpu_profiling.cc | 275 << ",\"kernel_name:" << event.kernel_name << "\",kernel_type:" << event.kernel_type in EventLog()
369 event.kernel_type = typestring; in EventHandleProcess()
604 profilingData->kernel_type = "cuMemcpy"; in HandleActivityMemcpyRecord()
621 profilingData->kernel_type = "cuMemcpy"; in HandleActivityMemcpy2Record()
654 profilingData->kernel_type = "cuLaunchKernel"; in HandleActivityKernelRecord()
|
| D | gpu_data_saver.h | 52 os << "\"" << event.basic_info_->kernel_name << "\"," << event.basic_info_->kernel_type << ','
|
| D | gpu_profiling.h | 72 std::string kernel_type; member
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/pass/ |
| D | common_subexpression_elimination_test.cc | 46 KernelType kernel_type) { in CreateKernelBuildInfo() argument
52 builder1.SetKernelType(kernel_type); in CreateKernelBuildInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | transpose_transdata_fusion.cc | 52 new_transdata_builder->SetKernelType(transdata_kernel_build_info->kernel_type()); in Process()
|
| D | parameter_and_transop_fusion.cc | 73 builder.SetKernelType(cast_build_info->kernel_type()); in GetKernelBuildInfo()
|
| /third_party/boost/boost/gil/extension/numeric/ |
| D | algorithm.hpp | 200 using kernel_type = typename std::iterator_traits<KernelIterator>::value_type; in correlate_pixels_k() typedef
212 pixel_multiplies_scalar_t<src_pixel_ref_t, kernel_type, PixelAccum>()), in correlate_pixels_k()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_helper.h | 87 void ResetKernelInfo(const AnfNodePtr &node, KernelType kernel_type = KernelType::UNKNOWN_KERNEL_TY…
|
| D | graph_kernel_helper.cc | 604 void ResetKernelInfo(const AnfNodePtr &node, KernelType kernel_type) { in ResetKernelInfo() argument
608 device::ascend::SetKernelInfo(cnode, kernel_type); in ResetKernelInfo()
611 device::gpu::SetKernelInfo(cnode, kernel_type); in ResetKernelInfo()
|
| /third_party/mindspore/mindspore/lite/src/train/ |
| D | train_session.cc | 346 auto kernel_type = kernel->desc().data_type; in MixPrecisionPreProcess() local
354 if (tensor->data_type() != kernel_type) { in MixPrecisionPreProcess()
355 …auto restore_tensor = CastTensor(tensor, kernel_type, this->context_->device_and_pkg_support_fp16(… in MixPrecisionPreProcess()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | ascend_kernel_compile.cc | 526 KernelType kernel_type = AnfAlgo::GetKernelType(anf_node); in GetAllAscendNodes() local
527 if (kernel_type == TBE_KERNEL) { in GetAllAscendNodes()
|
| /third_party/ltp/scripts/ |
| D | checkpatch.pl | 6643 my $kernel_type = 'u';
6644 $kernel_type = 's' if ($type =~ /^_*[si]/);
6646 $kernel_type .= $1;
6648 "Prefer kernel type '$kernel_type' over '$type'\n" . $herecurr) &&
6650 $fixed[$fixlinenr] =~ s/\b$type\b/$kernel_type/;
|
| /third_party/mindspore/third_party/proto/tensorflow/lite/ |
| D | schema.fbs | 618 kernel_type: LSTMKernelType = FULL;
|