Searched defs:arg_nodes (Results 1 – 6 of 6) sorted by relevance
208 Status CollectArgNodes(const Graph& graph, std::vector<Node*>* arg_nodes) { in CollectArgNodes()238 std::vector<Node*> arg_nodes; in CreateXlaArgs() local
115 std::unordered_map<int, Node*> arg_nodes; in ReplaceArgUsageWithConstNode() local
80 gtl::InlinedVector<Node*, 4> arg_nodes; member
77 std::vector<std::pair<Node*, int>> arg_nodes; in UpdateArgAndRetvalMetadata() local
75 Status GetArgDataTypes(const std::vector<Node*>& arg_nodes, in GetArgDataTypes()124 std::vector<Node*> arg_nodes = GatherNodesWithType(*g, "_Arg"); in ReplaceArgNodesWithRecvAtHostNode() local
787 TFNode** arg_nodes = builder_->Buffer(param_count + 1); in DoCall() local