Searched refs:kXlaTokenArgNodeName (Results 1 – 5 of 5) sorted by relevance
36 extern const char kXlaTokenArgNodeName[];
25 const char kXlaTokenArgNodeName[] = "_xla_token_arg_node"; variable
1346 SetNodeToken(kXlaTokenArgNodeName, arg_expression.handle())); in CompileGraph()1354 output_node_token_inputs.insert(kXlaTokenArgNodeName); in CompileGraph()1360 SetNodeToken(kXlaTokenArgNodeName, xla::CreateToken(&builder))); in CompileGraph()
1513 std::vector<string>{kXlaTokenArgNodeName}, &side_effecting_op); in TEST_F()
347 xla_token_input_nodes.emplace_back(kXlaTokenArgNodeName); in BuildXlaHostComputeNodeDef()1331 std::vector<string>{kXlaTokenArgNodeName}); in BuildSendIfPredNode()1514 std::vector<string>{kXlaTokenArgNodeName}); in AddSendLoopPredToLoopCond()1875 std::vector<string>{kXlaTokenArgNodeName}); in ExtractOutsideCompilationForFuncCallNode()2066 std::vector<string>{kXlaTokenArgNodeName}); in ExtractOutsideCompilationForWhileNode()2422 if (node_name == kXlaTokenArgNodeName) { in ExtractOutsideCompilationForFunction()