Home
last modified time | relevance | path

Searched defs:real_input_index (Results 1 – 16 of 16) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/
Ddynamic_shape_helper.cc114 auto real_input_index = input_node_with_index.second; in CreateTensorFromIndexedNode() local
427 auto real_input_index = input_node_with_index.second; in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dgrouped_matmul_info.cc280 size_t real_input_index = 0; in InferTensorInfo() local
Dquant_batch_matmul_info.cc491 size_t real_input_index = 0; in InferTensorLayout() local
Doperator_info.cc572 size_t real_input_index = 0; in InferTensorInfoNew() local
623 size_t real_input_index = 0; in InferTensorInfo() local
1501 size_t real_input_index = 0; in InitWithTensorLayout() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/oplib/
Dopinfo.h134 const std::pair<std::map<size_t, size_t>, std::map<size_t, size_t>> &real_input_index() const { in real_input_index() function
174 …nput_index(const std::pair<std::map<size_t, size_t>, std::map<size_t, size_t>> &real_input_index) { in set_real_input_index() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dops_utils.h192 const T real_input_index = AreaPixelComputeSourceIndex<T>(ratio, output_index, align_corners); in ComputeSourceIndexAndLambda() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/rpc/
Drecv_actor.cc288 size_t real_input_index = input_node_with_index.second; in AddArgSpecForInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dupsample_trilinear3d.cc91 const T real_input_index = AreaPixelComputeSourceIndex<T>(ratio, output_index, align_corners); in ComputeSourceIndexAndLambda() local
Dupsample_trilinear3d_grad.cc91 const T real_input_index = AreaPixelComputeSourceIndex<T>(ratio, output_index, align_corners); in ComputeSourceIndexAndLambda() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dresize_v2_grad_cpu_kernel.cc63 …const T real_input_index = AreaPixelComputeSourceIndex<T>(ratio, output_index, align_corners, fals… in ComputeSourceIndexAndLambda() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/
Dop_info_register.py786 def real_input_index(self, real_input_index): member in TBERegOp
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/mem_reuse/
Dmem_reuse_checker.cc409 auto real_input_index = AnfAlgo::GetInputGraphIdxByKernelIdx(node, i); in CheckNormalIR() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/graph_kernel/
Dgraph_kernel_json_generator.cc435 size_t real_input_index = 0; in CreateInputDescJson() local
1032 size_t real_input_index = 0; in UpdateTensorName() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dop_adapter.cc698 size_t real_input_index = i; in UpdateNormalOpInputDesc() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/somas/
Dsomas.cc828 size_t real_input_index = 0; in InitCommonNodeInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Danfalgo.cc1868 …gList(AbstractBasePtrList *args_spec_list, const AnfNodePtr &real_input, size_t real_input_index) { in AddArgList()