Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_communication.cc418 llvm::SmallDenseMap<Value, Value>& rewritten_values, bool flatten_tuple) { in GetValueWithToken() argument
419 if (flatten_tuple) { in GetValueWithToken()
466 bool flatten_tuple) { in GetTypeWithToken() argument
470 if (flatten_tuple) { in GetTypeWithToken()
505 ArrayRef<Value> replacements, bool flatten_tuple) { in ReplaceWithTupleResult() argument
506 if (flatten_tuple) { in ReplaceWithTupleResult()
567 Value token, bool flatten_tuple) { in RewriteControlFlowTerminator() argument
568 assert(flatten_tuple || terminator->getNumOperands() == 1); in RewriteControlFlowTerminator()
569 assert(flatten_tuple || terminator->getBlock()->getNumArguments() == 1); in RewriteControlFlowTerminator()
579 token, rewritten_operands, flatten_tuple); in RewriteControlFlowTerminator()
[all …]