Searched refs:node_token_mapping_stack_ (Results 1 – 2 of 2) sorted by relevance
1570 node_token_mapping_stack_.emplace(std::map<string, xla::XlaOp>{}); in PushNodeTokenMapping()1574 if (node_token_mapping_stack_.empty()) { in PopNodeTokenMapping()1579 node_token_mapping_stack_.pop(); in PopNodeTokenMapping()1585 if (node_token_mapping_stack_.empty()) { in SetNodeToken()1590 auto insert_result = node_token_mapping_stack_.top().insert({node_name, op}); in SetNodeToken()1599 if (node_token_mapping_stack_.empty()) { in GetNodeToken()1604 auto iter = node_token_mapping_stack_.top().find(node_name); in GetNodeToken()1605 if (iter == node_token_mapping_stack_.top().end()) { in GetNodeToken()
377 std::stack<std::map<string, xla::XlaOp>> node_token_mapping_stack_; variable