Home
last modified time | relevance | path

Searched defs: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()
466 bool flatten_tuple) { in GetTypeWithToken()
505 ArrayRef<Value> replacements, bool flatten_tuple) { in ReplaceWithTupleResult()
567 Value token, bool flatten_tuple) { in RewriteControlFlowTerminator()
876 bool flatten_tuple = isa<mhlo::WhileOp, mhlo::IfOp, mhlo::CaseOp>( in RewriteFunction() local