Home
last modified time | relevance | path

Searched defs:KernelWithIndex (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Deliminate_redundant_op.h32 using KernelWithIndex = std::pair<CNodePtr, size_t>; variable
Dcommon_subexpression_elimination.cc34 using KernelWithIndex = std::pair<AnfNodePtr, int64_t>; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dgraph_compiler.cc828 std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in GetParamAndOutputIndex()
835const std::map<KernelWithIndex, tensor::BaseTensorPtr> &op_output, in GetSingleOpInputTensors()
848 const CNodePtr &kernel, const std::map<KernelWithIndex, tensor::BaseTensorPtr> &op_output, in GetSingleOpInputTensorByIndex()
864 void GraphCompiler::CalculateRefCount(const KernelGraphPtr &graph, std::map<KernelWithIndex, size_t… in CalculateRefCount()
880 std::map<KernelWithIndex, size_t> *ref_count, in UpdateRefCount()
894 const std::map<KernelWithIndex, size_t> &ref_count, in RecoverGraphOutput()
Dany_type_graph_scheduler.cc516const std::map<KernelWithIndex, std::pair<AbstractActor *, KernelWithIndex>, in Optimize() argument
Dcontrol_node_scheduler.cc438 [&exit_actor, &graph](const std::pair<KernelWithIndex, KernelWithIndex> &pair) { in BuildExitActor()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dcallback_impl.h28 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; variable
Dformat_recognition.cc33 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dsession_basic.cc281std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in CreateNodeOutputPlaceholder()
314std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in CreateNodeOutputPlaceholder()
519 std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) const { in CreateOutputPlaceholder()
533 void SessionBasic::GetRefCount(const KernelGraph *graph, std::map<KernelWithIndex, size_t> *ref_cou… in GetRefCount()
543 void SessionBasic::CalculateRefCount(const AnfNodePtr &node, std::map<KernelWithIndex, size_t> *ref… in CalculateRefCount()
624 std::map<KernelWithIndex, size_t> *ref_count, in HandleOpInputs()
655 const std::map<KernelWithIndex, size_t> &ref_count, in HandleOpOutputs()
774 …const KernelWithIndex &kernel_with_index, const std::map<KernelWithIndex, tensor::BaseTensorPtr> &… in GetCNodeOutputTensor() argument
821const std::map<KernelWithIndex, tensor::BaseTensorPtr> &op_output, in GetOpInputTensors()
904const std::map<KernelWithIndex, tensor::BaseTensorPtr> &op_output, in GetOpInputTensorsFromCNode()
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/
Dgllo_utils.h62 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/
Dbackend_base.h46 using KernelWithIndex = session::KernelWithIndex; variable
Dbackend.cc341 const std::map<KernelWithIndex, tensor::BaseTensorPtr> &op_output, in GetInputofBpropCut()
397 const std::map<KernelWithIndex, tensor::BaseTensorPtr> &op_output_map, in GetControlOpInput()
429 const std::map<KernelWithIndex, tensor::BaseTensorPtr> &op_output_map, in RunControlOperator()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/ascend_native/
Dascend_native_base_kernel.h32 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/utils/
Danfalgo.h46 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/
Dsplit_n_optimizer.cc33 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/
Dset_fracz_group_attr.cc35 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dop_compiler.cc39 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/
Dkernel_graph.h42 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; variable
Ddevice_address.h68 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; variable