Searched refs:GetValueWithToken (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 417 Value GetValueWithToken(OpBuilder& builder, Value value, Value token, in GetValueWithToken() function
540 Value new_result = GetValueWithToken(builder, terminator->getOperand(0), in RewriteControlFlowTerminator()
553 Value new_true_operand = GetValueWithToken(builder, region_if.true_arg(), in RewriteRegionIfOp()
555 Value new_false_operand = GetValueWithToken(builder, region_if.false_arg(), in RewriteRegionIfOp()
651 GetValueWithToken(builder, region_while.val(), token, rewritten_operands); in RewriteRegionWhileOp()
|