Searched defs:prefer_tf2xla (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | xla_legalize_tf.cc | 53 bool prefer_tf2xla) { in LegalizeTF()
288 bool prefer_tf2xla) { in legalizeTF()
383 llvm::Optional<StringRef> tf2xla_fallback_device_type, bool prefer_tf2xla) { in createLegalizeTFPass()
|
| D | legalize_tf_with_tf2xla.cc | 723 bool prefer_tf2xla, bool legalize_test_only_ops, in Tf2XlaRewritePattern()
759 explicit LegalizeTF(llvm::StringRef device_type, bool prefer_tf2xla) { in LegalizeTF()
783 MLIRContext* ctx, bool prefer_tf2xla, in PopulateLegalizeTfWithTf2XlaPatterns()
791 llvm::StringRef device_type, bool prefer_tf2xla) { in createLegalizeTfWithTf2XlaPass()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 377 mlir::OpPassManager& pm, llvm::StringRef device_type, bool prefer_tf2xla, in CreateConvertMlirToXlaHloPipeline()
461 bool prefer_tf2xla, in LegalizeToHlo()
519 bool prefer_tf2xla, bool return_tuple, in ConvertMLIRToXlaComputation()
|