Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Daddn_fission.cc38 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.cc41 std::vector<int64_t> dyn_input_sizes{SizeToLong(offset)}; in CreateNewPack() local
102 std::vector<int64_t> dyn_input_sizes{SizeToLong(base_concat_inputs.size() - 1)}; in Process() local
Dconcat_fission.cc46 std::vector<int64_t> dyn_input_sizes{SizeToLong(offset), (int64_t)-1}; in CreateNewConcat() local
127 std::vector<int64_t> dyn_input_sizes{SizeToLong(base_concat_inputs.size() - 1), (int64_t)-1}; in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/add_attr_to_node/
Dparallel_concat_fusion.cc43 std::vector<int64_t> dyn_input_sizes{n}; in ParallelConcatFusionProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/aclnn/
Dfused_infer_attention_score_aclnn_kernel.cc31 std::vector<int64_t> dyn_input_sizes; in GetWorkSpaceInfo() local
85 std::vector<int64_t> dyn_input_sizes; in Launch() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dgrouped_matmul_op_pass.cc97 … const size_t input_idx, std::vector<int64_t> *dyn_input_sizes, in UseEmptyNodeReplaceNone()
188 std::vector<int64_t> dyn_input_sizes; in ConvertMakeTupleInputToPlantInputs() local
Dconcat_op_pass.cc142 std::vector<int64_t> dyn_input_sizes; in ConvertMakeTupleInputToPlantInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/graph_kernel/
Dgraph_kernel_json_generator.cc53 std::vector<int64_t> dyn_input_sizes; in GetDynInputSizes() local
98 auto dyn_input_sizes = GetDynInputSizes(anf_node); in GetInputIndex() local
184 auto dyn_input_sizes = GetDynInputSizes(cnode_); in ExtractInputs() local
434 auto dyn_input_sizes = GetDynInputSizes(anf_node); in CreateInputDescJson() local
507 …JsonGenerator::GetAttrJson(const AnfNodePtr &anf_node, const std::vector<int64_t> &dyn_input_sizes, in GetAttrJson()
568 auto dyn_input_sizes = GetDynInputSizes(anf_node); in CreateAttrDescJson() local
1029 auto dyn_input_sizes = GetDynInputSizes(anf_node); in UpdateTensorName() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dgrouped_matmul.cc140 std::vector<int64_t> dyn_input_sizes; in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dflatten_concat_fission.cc135 std::vector<int64_t> dyn_input_sizes; in ExpandFlattenConcatTupleInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dframework_utils.cc317 size_t builder_idex, const std::vector<int64_t> &dyn_input_sizes, in SetInputKernelBuilderInfo()
500 std::vector<int64_t> dyn_input_sizes; in ParseMetadata() local
982 std::vector<int64_t> dyn_input_sizes; in SetDynamicInputSizeAttr() local
Dcommon_utils.cc516 const std::vector<int64_t> &dyn_input_sizes, in MatchMultiDynamicKernelAttr()
764 std::vector<int64_t> dyn_input_sizes = {}; in SetCpuRefMapToKernelInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dkernel_select_cpu.cc224 …id ExpandMultiDynamicAttr(const CNodePtr &kernel_node, const std::vector<int64_t> &dyn_input_sizes, in ExpandMultiDynamicAttr()
285 …auto dyn_input_sizes = common::AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, kAttrDynInp… in ExpandKernelAttrByDynamicSize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/
Dir_pass.cc478 std::vector<int64_t> dyn_input_sizes; in ConvertMakeTupleInputToDynamicInput() local
655 …const auto &dyn_input_sizes = common::AnfAlgo::GetNodeAttr<std::vector<int64_t>>(cnode, kAttrDynIn… in ReverseMakeTupleNode() local
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/ops/
Dprimitive_infer_map.cc112 std::vector<int64_t> dyn_input_sizes = GetValue<std::vector<int64_t>>(attr); in RectifyDependListFromDynamicInputAttr() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/
Dbase.h59 std::vector<int64_t> dyn_input_sizes; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/
Dacl_convert.cc778 std::vector<int64_t> dyn_input_sizes = {}; in ConvertMsIdxToGeIdx() local
907 std::vector<int64_t> dyn_input_sizes = {}; in ResizeAclOpInputs() local
Dacl_helper.cc323 std::vector<int64_t> dyn_input_sizes = {}; in GetKernelInfoByInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/
Dfunc_builder.cc66 std::vector<int64_t> dyn_input_sizes = GetValue<std::vector<int64_t>>(attr); in GetValueDependArgIndices() local
Dfunc_pass.cc256 std::vector<int64_t> dyn_input_sizes; in ConvertMakeTupleInputToDynamicInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dconvert_pad_v3_paddings.cc108 std::vector<int64_t> dyn_input_sizes = {SizeToLong(concat_input_vec.size()), -1}; in CreateConcatNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/
Dhelper.cc1468std::vector<int64_t> *dyn_input_sizes, std::vector<AnfNodePtr> *plant_inputs) { in UseEmptyNodeReplaceNone()
1489 std::vector<int64_t> *dyn_input_sizes) { in GetPlantInputsAndSize()
1551 std::vector<int64_t> dyn_input_sizes; in ConvertMakeTupleInputToPlantInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dconvert.cc287 int64_t GetDynInputNum(const OpAdapterPtr &adpt, bool is_call, std::vector<int64_t> dyn_input_sizes, in GetDynInputNum()
391const std::vector<int64_t> &dyn_input_sizes, const int &ge_input_size, in GetGeTensorOrders()
2951 std::vector<int64_t> *dyn_input_sizes) { in SetDynamicInputBeforeNormalInput()
3097 std::vector<int64_t> dyn_input_sizes; in SetOpInput() local
Dop_adapter.cc681 …std::vector<int64_t> dyn_input_sizes = common::AnfAlgo::GetNodeAttr<std::vector<int64_t>>(node, kA… in UpdateNormalOpInputDesc() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dflash_attention_fusion.cc1613 std::vector<int64_t> dyn_input_sizes = {-1, 1, 1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1}; in CreateIncreFlashAttentionCnodeForBNSD() local