Searched refs:output_tensor_num (Results 1 – 3 of 3) sorted by relevance
38 size_t output_tensor_num = AnfAlgo::GetOutputTensorNum(kernel_node); in FilterInvalidKernelInfo() local43 …[output_tensor_num, input_tensor_num](const std::shared_ptr<kernel::KernelBuildInfo> &kernel_build… in FilterInvalidKernelInfo()44 return kernel_build_info->GetOutputNum() == output_tensor_num && in FilterInvalidKernelInfo()55 if (kernel_info->GetOutputNum() != output_tensor_num) { in FilterInvalidKernelInfo()56 buffer << "Kernel node's output size [" << output_tensor_num << "]" in FilterInvalidKernelInfo()65 …INFO) << "Node: " << kernel_node->DebugString() << "'s output size : [" << output_tensor_num << "]" in FilterInvalidKernelInfo()
80 size_t output_tensor_num = AnfAlgo::GetOutputTensorNum(cnode_); in ExtractOutputs() local81 for (size_t i = 0; i < output_tensor_num; i++) { in ExtractOutputs()229 size_t output_tensor_num = AnfAlgo::GetOutputTensorNum(anf_node); in CreateOutputDescJson() local232 for (size_t i = 0; i < output_tensor_num; i++) { in CreateOutputDescJson()
1523 auto output_tensor_num = GetOutputTensorNum(preceed_node); in Run() local1524 if (output_tensor_num > 1) { in Run()