Searched refs:node_token_mapping_stack_ (Results 1 – 2 of 2) sorted by relevance
1294 node_token_mapping_stack_.emplace(std::map<string, xla::XlaOp>{}); in PushNodeTokenMapping()1298 if (node_token_mapping_stack_.empty()) { in PopNodeTokenMapping()1303 node_token_mapping_stack_.pop(); in PopNodeTokenMapping()1309 if (node_token_mapping_stack_.empty()) { in SetNodeToken()1314 auto insert_result = node_token_mapping_stack_.top().insert({node_name, op}); in SetNodeToken()1323 if (node_token_mapping_stack_.empty()) { in GetNodeToken()1328 auto iter = node_token_mapping_stack_.top().find(node_name); in GetNodeToken()1329 if (iter == node_token_mapping_stack_.top().end()) { in GetNodeToken()
492 std::stack<std::map<string, xla::XlaOp>> node_token_mapping_stack_; variable