Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.h76 xla::XlaComputation* xla_computation, bool use_tuple_args,
108 bool use_tuple_args, bool use_resource_updates_for_aliases,
119 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph,
130 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph,
143 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph,
155 bool use_tuple_args, bool analyse_graph,
Dcompile_mlir_util.cc85 bool use_tuple_args, in GetXlaInputShapes() argument
125 if (use_tuple_args) { in GetXlaInputShapes()
410 xla::XlaComputation* xla_computation, bool use_tuple_args, in ConvertMLIRToXlaComputation() argument
420 use_tuple_args, return_tuple, in ConvertMLIRToXlaComputation()
468 bool use_tuple_args, bool use_resource_updates_for_aliases, in PopulateResultIOInfo() argument
476 TF_RETURN_IF_ERROR(GetXlaInputShapes(module_op, arg_shapes, use_tuple_args, in PopulateResultIOInfo()
489 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileMlirToXlaHlo() argument
508 use_tuple_args, analyse_graph, use_return_tuple, shape_representation_fn, in CompileMlirToXlaHlo()
511 return PopulateResultIOInfo(module_op, arg_shapes, use_tuple_args, in CompileMlirToXlaHlo()
518 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileSerializedMlirToXlaHlo() argument
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.h55 bool use_tuple_args, bool return_tuple,
Dmlir_hlo_to_hlo.cc469 bool use_tuple_args, bool return_tuple, in ConvertToHloModule() argument
474 use_tuple_args_(use_tuple_args), in ConvertToHloModule()
1785 bool use_tuple_args) { in AddDynamicParameterBindingEntry() argument
1788 if (use_tuple_args) { in AddDynamicParameterBindingEntry()
1824 mlir::ModuleOp module, xla::HloProto* hlo_proto, bool use_tuple_args, in ConvertMlirHloToHlo() argument
1831 ConvertToHloModule converter(module, module_builder, use_tuple_args, in ConvertMlirHloToHlo()