Searched defs:input_with_index (Results 1 – 6 of 6) sorted by relevance
82 auto input_with_index = AnfAlgo::GetPrevNodeOutput(kernel, i); in GetNeedDumpKernelList() local
150 auto input_with_index = AnfAlgo::VisitKernel(input_kernel_node, 0); in SetTensorDeviceInfo() local
440 auto input_with_index = AnfAlgo::VisitKernelWithReturnType(input_kernel_node, 0); in SetTensorDeviceInfo() local
597 auto input_with_index = AnfAlgo::GetPrevNodeOutput(kernel, i); in UpdateNeedDumpKernels() local
1711 const auto &input_with_index = AnfAlgo::VisitKernelWithReturnType(inputs[i], 0); in LinkArrowByControlNode() local1744 const auto &input_with_index = gather_actor->data_nodes_[i]; in LinkArrowByControlNode() local1833 const KernelWithIndex &input_with_index, in LinkDataArrowByControlNode()
2074 auto input_with_index = AnfAlgo::GetPrevNodeOutput(anf_node_ptr, i); in IsNodeInputDynamicShape() local2229 auto input_with_index = AnfAlgo::GetPrevNodeOutput(node, i); in InferShape() local2337 auto input_with_index = GetPrevNodeOutput(node, i); in IsNodeInputContainMonad() local