Searched refs:CreateSendOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 254 Value CreateSendOp(OpBuilder& builder, int64_t& channel_id, Location loc, in CreateSendOp() function
336 CreateSendOp(builder, channel_id, loc, operand.value(), in RewriteHostComputeOp()
359 token = CreateSendOp(builder, channel_id, send_to_host.getLoc(), in RewriteSendToHostOp()
|