Searched defs:real_input_index (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | add_placeholder_for_dynamic_gru.cc | 52 size_t real_input_index = 0; in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | single_tbe_json_creator.cc | 147 void SingleTbeJsonCreator::GenInputDescJson(const AnfNodePtr &anf_node, size_t real_input_index, in GenInputDescJson()
349 void SelectTbeJsonCreator::GenInputDescJson(const AnfNodePtr &anf_node, size_t real_input_index, in GenInputDescJson()
402 void CheckTbeJsonCreator::GenInputDescJson(const AnfNodePtr &anf_node, size_t real_input_index, in GenInputDescJson()
|
| D | tbe_json_creator.cc | 498 void TbeJsonCreator::GenInputConstValue(const AnfNodePtr &anf_node, size_t real_input_index, in GenInputConstValue()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 181 auto real_input_index = AnfAlgo::GetRealInputIndex(anf_node_ptr, i); in LaunchKernel() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_build.cc | 284 …onCreator::GenValidInputDescJson(const std::shared_ptr<AnfNode> &anf_node, size_t real_input_index, in GenValidInputDescJson()
316 …nelJsonCreator::GenInputDescJson(const std::shared_ptr<AnfNode> &anf_node, size_t real_input_index, in GenInputDescJson()
343 … const std::shared_ptr<OpIOInfo> &input_ptr, size_t *real_input_index, in GenInputList()
436 size_t real_input_index = 0; in GenTbeInputsJson() local
|
| D | tbe_adapter.cc | 256 … size_t real_input_index, std::vector<nlohmann::json> *input_list, in GenTopKV2IndicesTensorInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_generator.cc | 182 size_t real_input_index = 0; in CreateInputDescJson() local
685 size_t real_input_index = 0; in UpdateTensorName() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_reuse_checker.cc | 381 auto real_input_index = AnfAlgo::GetRealInputIndex(node, i); in CheckNormalIR() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | clean.cc | 291 constexpr size_t real_input_index = 1; in ConvertListGetItemToTupleGetItem() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 515 size_t real_input_index = 0; in InitCommonNodeInputs() local
|