Home
last modified time | relevance | path

Searched refs:is_feature_map (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_info.h72 bool is_feature_map() const { return is_feature_map_; } in is_feature_map() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_decoder.cc254 AnfAlgo::SetNodeAttr(kIsFeatureMapOutput, MakeValue(kernel_info->is_feature_map()), cnode_); in CreateKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_helper.cc719 SetNodeAttrSafely(kIsFeatureMapOutput, MakeValue(kernel_info->is_feature_map()), cnode); in CreateCNode()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dkernel_graph.cc523 AnfAlgo::SetNodeAttr(kIsFeatureMapOutput, MakeValue(kernel_info->is_feature_map()), node); in SetKernelInfoForNode()
Dascend_session.cc217 output_tensor_info.is_weight = !(kernel_info->is_feature_map()); in GenOpOutputStubTensor()
Danf_runtime_algorithm.cc1564 return kernel_info->is_feature_map(); in IsFeatureMapOutput()