Home
last modified time | relevance | path

Searched defs:dyn_input_sizes (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Daddn_fission.cc37 std::vector<int64_t> dyn_input_sizes{SizeToLong(offset)}; in CreateNewAddn() local
76 std::vector<int64_t> dyn_input_sizes{SizeToLong(base_addn_inputs.size() - 1)}; in Process() local
Dpack_fission.cc40 std::vector<int64_t> dyn_input_sizes{SizeToLong(offset)}; in CreateNewPack() local
100 std::vector<int64_t> dyn_input_sizes{SizeToLong(base_concat_inputs.size() - 1)}; in Process() local
Dconcat_fission.cc45 std::vector<int64_t> dyn_input_sizes{SizeToLong(offset)}; in CreateNewConcat() local
117 std::vector<int64_t> dyn_input_sizes{SizeToLong(base_concat_inputs.size() - 1)}; in Process() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_generator.cc31 std::vector<int> dyn_input_sizes; in GetDynInputSize() local
62 auto dyn_input_sizes = GetDynInputSize(cnode_); in ExtractInputs() local
181 auto dyn_input_sizes = GetDynInputSize(anf_node); in CreateInputDescJson() local
256 …rnelJsonGenerator::GetAttrJson(const AnfNodePtr &anf_node, const std::vector<int> &dyn_input_sizes, in GetAttrJson()
305 auto dyn_input_sizes = GetDynInputSize(anf_node); in CreateAttrDescJson() local
682 auto dyn_input_sizes = GetDynInputSize(anf_node); in UpdateTensorName() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconvert_tuple_input_to_dynamic_input.cc67 std::vector<int64_t> dyn_input_sizes; in ConvertMakeTupleInputToPlantInputs() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc93 auto dyn_input_sizes = GetNodeDynamicInputs(); in GetCommonPatternKernelInfo() local
309 std::vector<int64_t> dyn_input_sizes; in GetNodeDynamicInputs() local
318const std::vector<int64_t> &dyn_input_sizes, std::vector<std::string> *formats, in GenBuilderItem()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.cc301 size_t builder_idex, const std::vector<int64_t> &dyn_input_sizes, in SetInputKernelBuilderInfo()
437 std::vector<int64_t> dyn_input_sizes; in ParseMetadata() local
605 std::vector<int64_t> dyn_input_sizes; in GetInputIndex() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dfusion_tbe_json_creator.cc228 auto dyn_input_sizes = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(cnode, kAttrDynInputSizes); in CheckDynamicInput() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_build.cc399 std::vector<int64_t> dyn_input_sizes; in GetInputNameAndRealNum() local
1304 std::vector<int64_t> dyn_input_sizes; in IsDynamicInput() local