Home
last modified time | relevance | path

Searched refs:CreateSinkToken (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_communication.cc310 Value CreateSinkToken(OpBuilder& builder, Location loc, ArrayRef<Value> tokens, in CreateSinkToken() function
340 token = CreateSinkToken(builder, loc, send_tokens, token); in RewriteHostComputeOp()
349 token = CreateSinkToken(builder, loc, recv_tokens, token); in RewriteHostComputeOp()