Searched defs:token_output (Results 1 – 5 of 5) sorted by relevance
339 xla::XlaOp token_output = in Compile() local
340 xla::XlaOp token_output = in Compile() local
603 xla::XlaOp token_output = in Compile() local
247 xla::XlaOp token_output = recv_from_host_tokens.empty() in Compile() local
170 std::unique_ptr<xla::XlaOp> token_output, in BuildComputation()1329 std::unique_ptr<xla::XlaOp> token_output; in CompileGraph() local