Searched refs:send_to_host (Results 1 – 2 of 2) sorted by relevance
357 TF::XlaSendToHostOp send_to_host, Value token) { in RewriteSendToHostOp() argument358 builder.setInsertionPoint(send_to_host); in RewriteSendToHostOp()359 token = CreateSendOp(builder, channel_id, send_to_host.getLoc(), in RewriteSendToHostOp()360 send_to_host.input(), send_to_host.key(), in RewriteSendToHostOp()363 send_to_host.erase(); in RewriteSendToHostOp()773 } else if (auto send_to_host = dyn_cast<TF::XlaSendToHostOp>(curr_op)) { in RewriteFunction() local774 token = RewriteSendToHostOp(builder, channel_id, send_to_host, token); in RewriteFunction()
164 // CHECK-LABEL: func @send_to_host166 func @send_to_host(%arg0: tensor<i32>) {