Home
last modified time | relevance | path

Searched defs:real_input_num (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Drandom_op_gpu_kernel.cc55 auto real_input_num = inputs.size(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dlamb_fission.cc196 auto real_input_num = common::AnfAlgo::GetInputNum(lamb_cnode); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dlamb_fission.cc249 auto real_input_num = common::AnfAlgo::GetInputNum(lamb_cnode); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dframework_utils.cc316 …InputKernelBuilderInfo(const std::vector<std::shared_ptr<OpIOInfo>> &inputs, size_t real_input_num, in SetInputKernelBuilderInfo()
496 size_t real_input_num = AnfAlgo::GetInputElementNum(kernel_node); in ParseMetadata() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dsuper_kernel_actor.cc905 auto real_input_num = common::AnfAlgo::GetInputTensorNum(kernel); in ParseInputIndex() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dkernel_select_ascend.cc193 size_t real_input_num = 0; in SetTensorDeviceInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Danfalgo.cc1683 size_t real_input_num = cnode->size() - 1; // exclude primitive in input[0] in IsDynamicValue() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Danf_runtime_algorithm.cc1689 auto real_input_num = common::AnfAlgo::GetInputTensorNum(kernel); in HasComputedDependInputNode() local