Searched defs:input_tensor_num (Results 1 – 11 of 11) sorted by relevance
110 size_t input_tensor_num = inputs.size(); in Resize() local184 …t64_t> PrintGpuKernelMod::SetInputFlag(std::vector<int64_t> *string_pos, size_t input_tensor_num) { in SetInputFlag()
72 size_t input_tensor_num = inputs.size() - kConcatOffsetV1AxisNum; in LaunchKernel() local
49 size_t input_tensor_num = common::AnfAlgo::GetInputTensorNum(real_node); in InitKernel() local
92 const vector<ShapeVector> &shape_list, const size_t input_tensor_num, in GetHcomCount()
59 …PrintErrorInfo(const AnfNodeWeakPtrList &inputs, const PrimitivePtr &prim, size_t input_tensor_num, in PrintErrorInfo()88 size_t input_tensor_num = inputs.size() - prim_num; in CheckCNodeInputsNum() local568 ssize_t CNode::input_tensor_num() const { return input_tensor_num_; } in input_tensor_num() function in mindspore::CNode600 void CNode::set_input_tensor_num(ssize_t input_tensor_num) { input_tensor_num_ = input_tensor_num; } in set_input_tensor_num()
114 auto input_tensor_num = common::AnfAlgo::GetInputTensorNum(node); in AssignStream() local237 auto input_tensor_num = common::AnfAlgo::GetInputTensorNum(kernel); in ProcessSideEffect() local
185 size_t input_tensor_num = common::AnfAlgo::GetInputTensorNum(kernel); in SetInputMap() local
267 ssize_t input_tensor_num = cnode->input_tensor_num(); in GetInputTensorNum() local
443 size_t input_tensor_num = dyn_input_sizes.empty() ? 1 : LongToSize(dyn_input_sizes[i]); in CreateInputDescJson() local1034 size_t input_tensor_num = is_dynamic_input ? LongToSize(dyn_input_sizes[i]) : 1; in UpdateTensorName() local
827 auto input_tensor_num = common::AnfAlgo::GetInputTensorNum(kernel); in InitCommonNodeInputs() local900 auto input_tensor_num = common::AnfAlgo::GetInputTensorNum(kernel); in InitAtomicCleanInputs() local
571 const auto input_tensor_num = common::AnfAlgo::GetInputTensorNum(kernel); in GetForwardOpOutputRefCount() local