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
1109 SetNodeToken(kXlaTokenArgNodeName, arg_expression.handle())); in CompileGraph()1117 output_node_token_inputs.insert(kXlaTokenArgNodeName); in CompileGraph()1123 SetNodeToken(kXlaTokenArgNodeName, xla::CreateToken(&builder))); in CompileGraph()
1447 std::vector<string>{kXlaTokenArgNodeName}, &side_effecting_op); in TEST_F()
314 std::vector<string>{kXlaTokenArgNodeName}); in BuildXlaHostComputeNodeDef()812 std::vector<string>{kXlaTokenArgNodeName}); in BuildSendIfPredNode()1002 std::vector<string>{kXlaTokenArgNodeName}); in AddSendLoopPredToLoopCond()1340 std::vector<string>{kXlaTokenArgNodeName}); in ExtractOutsideCompilationForNodesWithAssociatedFunctions()1465 std::vector<string>{kXlaTokenArgNodeName}); in ExtractOutsideCompilationForNodesWithAssociatedFunctions()