Searched defs:input_tensor_num (Results 1 – 12 of 12) sorted by relevance
104 size_t input_tensor_num = AnfAlgo::GetInputTensorNum(kernel_node); in Init() local191 std::vector<int64_t> SetInputFlag(std::vector<int64_t> *string_pos, size_t input_tensor_num) { in SetInputFlag()
29 size_t input_tensor_num = AnfAlgo::GetInputTensorNum(kernel_node); in InitKernel() local
39 size_t input_tensor_num = AnfAlgo::GetInputTensorNum(kernel_node); in FilterInvalidKernelInfo() local
104 auto input_tensor_num = AnfAlgo::GetInputTensorNum(anf_node_ptr); in LaunchAddrCleanKernel() local
312 ssize_t input_tensor_num() const { return input_tensor_num_; } in input_tensor_num() function352 void set_input_tensor_num(ssize_t input_tensor_num) { input_tensor_num_ = input_tensor_num; } in set_input_tensor_num()
190 size_t input_tensor_num = dyn_input_sizes.empty() ? 1 : IntToSize(dyn_input_sizes[i]); in CreateInputDescJson() local687 size_t input_tensor_num = is_dynamic_input ? IntToSize(dyn_input_sizes[i]) : 1; in UpdateTensorName() local
186 size_t input_tensor_num = AnfAlgo::GetInputTensorNum(kernel); in SetInputMap() local
514 auto input_tensor_num = AnfAlgo::GetInputTensorNum(kernel); in InitCommonNodeInputs() local586 auto input_tensor_num = AnfAlgo::GetInputTensorNum(kernel); in InitAtomicCleanInputs() local
342 …eKernelJsonCreator::GenInputList(const std::shared_ptr<AnfNode> &anf_node, size_t input_tensor_num, in GenInputList()439 size_t input_tensor_num; in GenTbeInputsJson() local
1086 const auto input_tensor_num = AnfAlgo::GetInputTensorNum(cnode); in GetOpInputStubTensors() local
591 ssize_t input_tensor_num = cnode->input_tensor_num(); in GetInputTensorNum() local
1439 const auto input_tensor_num = AnfAlgo::GetInputTensorNum(cnode); in GetOpInputTensors() local