Home
last modified time | relevance | path

Searched defs:dynamic_input_index (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/akg/
Dcpu_kernel_builder.cc41 std::string dynamic_input_index = GetValue<std::string>(fg->get_attr("dynamic_input_index")); in CreateCustomOp() local
84 std::string dynamic_input_index = GetCNodeDynamicInputIndex(node->cast<CNodePtr>()); in CompileJsonsInAnfnodes() local
Dutils.cc96 std::string dynamic_input_index; in GetCNodeDynamicInputIndex() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/common/
Dutils.cc61 void GetCustomIndex(const std::string &dynamic_input_index, std::vector<size_t> *index) { in GetCustomIndex()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/config_parser/
Dconfig_file_parser.cc72 int DynBatchOrDynImage(const mindspore::ProfileConfigs &profile, size_t dynamic_input_index) { in DynBatchOrDynImage()
187 size_t dynamic_input_index = 0; in SetDynParams() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/
Dhelper.cc983 auto dynamic_input_index = GetValue<std::vector<int64_t>>(dyn_input_list); in RectifyAbstractFromDynamicInput() local