Searched refs:node_token_mapping_stack_ (Results 1 – 2 of 2) sorted by relevance
1534 node_token_mapping_stack_.emplace(std::map<string, xla::XlaOp>{}); in PushNodeTokenMapping()1538 if (node_token_mapping_stack_.empty()) { in PopNodeTokenMapping()1543 node_token_mapping_stack_.pop(); in PopNodeTokenMapping()1549 if (node_token_mapping_stack_.empty()) { in SetNodeToken()1554 auto insert_result = node_token_mapping_stack_.top().insert({node_name, op}); in SetNodeToken()1563 if (node_token_mapping_stack_.empty()) { in GetNodeToken()1568 auto iter = node_token_mapping_stack_.top().find(node_name); in GetNodeToken()1569 if (iter == node_token_mapping_stack_.top().end()) { in GetNodeToken()
365 std::stack<std::map<string, xla::XlaOp>> node_token_mapping_stack_; variable