Searched defs:token_output (Results 1 – 5 of 5) sorted by relevance
336 xla::XlaOp token_output = in Compile() local
321 xla::XlaOp token_output = in Compile() local
594 xla::XlaOp token_output = in Compile() local
235 xla::XlaOp token_output = recv_from_host_tokens.empty() in Compile() local
170 std::unique_ptr<xla::XlaOp> token_output, in BuildComputation()1317 std::unique_ptr<xla::XlaOp> token_output; in CompileGraph() local