Searched defs:use_tuple_args (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | compile_mlir_util.cc | 117 bool use_tuple_args, in GetXlaInputShapes()
518 xla::XlaComputation* xla_computation, bool use_tuple_args, in ConvertMLIRToXlaComputation()
616 bool use_tuple_args, bool use_resource_updates_for_aliases, in PopulateResultIOInfo()
637 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileMlirToXlaHlo()
667 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileSerializedMlirToXlaHlo()
821 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileGraphToXlaHlo()
878 bool use_tuple_args, bool analyse_graph, in CompileGraphToXlaHlo()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | mlir_to_hlo.cc | 36 bool use_tuple_args, bool return_tuple) { in MlirToXlaComputation()
94 bool use_tuple_args, bool return_tuple) { in ParseMlirModuleStringAndConvertToXlaComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | mlir.cc | 59 bool use_tuple_args, in PyMlirModuleToXlaComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_compile.cc | 380 bool use_tuple_args, XlaCompiler::CompilationResult* compilation_result) { in CompileTFFunctionToHlo()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | mlir_hlo_to_hlo.cc | 571 bool use_tuple_args, bool return_tuple, in ConvertToHloModule()
2525 bool use_tuple_args) { in AddDynamicParameterBindingEntry()
2564 bool use_tuple_args, bool return_tuple, in ConvertMlirHloToHlo()
|