Searched refs:forward_name (Results 1 – 2 of 2) sorted by relevance
50 bool IsGradNode(const string& name, string* forward_name) { in IsGradNode() argument64 *forward_name = name.substr(start, len); in IsGradNode()401 string forward_name; in AddNode() local402 if (IsGradNode(node->name(), &forward_name)) { in AddNode()403 auto grad_nodes_it = grad_nodes_.find(forward_name); in AddNode()408 std::pair<string, std::vector<TFGraphNode*>>(forward_name, {node})); in AddNode()441 const string& forward_name = it.first; in Build() local442 auto forward_it = forward_nodes_.find(forward_name); in Build()
92 def unique_grad_fn_name(forward_name): argument93 return "%s_grad_%s" % (forward_name, ops.uid())