Searched refs:token_input_nodes (Results 1 – 3 of 3) sorted by relevance
85 std::vector<string> token_input_nodes; in CalculateTokenInputsForOutputToken() local87 &token_input_nodes) in CalculateTokenInputsForOutputToken()89 token_input_nodes.empty()) { in CalculateTokenInputsForOutputToken()115 std::vector<string> token_input_nodes; in HasSideEffectingNodes() local116 if (GetNodeAttr(n->attrs(), kXlaTokenInputNodesAttrName, &token_input_nodes) in HasSideEffectingNodes()118 !token_input_nodes.empty()) { in HasSideEffectingNodes()
283 std::vector<string> token_input_nodes; in CompileFunctionalNode() local286 &token_input_nodes)); in CompileFunctionalNode()288 for (const string& node_name : token_input_nodes) { in CompileFunctionalNode()
901 std::vector<string> token_input_nodes; in TEST_F() local903 "_xla_token_input_nodes", &token_input_nodes)); in TEST_F()906 EXPECT_EQ(token_input_nodes, expected_token_input_nodes_0); in TEST_F()907 token_input_nodes.clear(); in TEST_F()911 "_xla_token_input_nodes", &token_input_nodes)); in TEST_F()912 EXPECT_EQ(token_input_nodes, expected_token_input_nodes_1); in TEST_F()983 std::vector<string> token_input_nodes; in TEST_F() local985 "_xla_token_input_nodes", &token_input_nodes)); in TEST_F()988 EXPECT_EQ(token_input_nodes, expected_token_input_nodes_0); in TEST_F()989 token_input_nodes.clear(); in TEST_F()[all …]