Searched refs:CreateRecvOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 275 Value CreateRecvOp(OpBuilder& builder, int64_t& channel_id, Location loc, in CreateRecvOp() function
345 CreateRecvOp(builder, channel_id, loc, result.value(), in RewriteHostComputeOp()
371 token = CreateRecvOp(builder, channel_id, recv_from_host.getLoc(), in RewriteRecvFromHostOp()
|