Searched refs:recv_from_host (Results 1 – 2 of 2) sorted by relevance
369 TF::XlaRecvFromHostOp recv_from_host, Value token) { in RewriteRecvFromHostOp() argument370 builder.setInsertionPoint(recv_from_host); in RewriteRecvFromHostOp()371 token = CreateRecvOp(builder, channel_id, recv_from_host.getLoc(), in RewriteRecvFromHostOp()372 recv_from_host.output(), recv_from_host.key(), in RewriteRecvFromHostOp()375 recv_from_host.erase(); in RewriteRecvFromHostOp()775 } else if (auto recv_from_host = dyn_cast<TF::XlaRecvFromHostOp>(curr_op)) { in RewriteFunction() local776 token = RewriteRecvFromHostOp(builder, channel_id, recv_from_host, token); in RewriteFunction()
182 // CHECK-LABEL: func @recv_from_host183 func @recv_from_host() -> tensor<i32> {