Searched refs:module_builder (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tools/api/generator/ |
| D | create_python_api.py | 316 def add_nested_compat_imports(module_builder, compat_api_versions, argument
330 imported_modules = module_builder.get_destination_modules()
338 module_builder.copy_imports(compat_module, subcompat_module)
339 module_builder.copy_imports('%s.compat' % compat_module,
366 module_builder.add_import(
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 444 mlir::ModuleOp module, xla::XlaBuilder& module_builder, in ConvertToHloModule() argument
449 module_builder_(module_builder), in ConvertToHloModule()
1758 xla::XlaBuilder module_builder("main"); in ConvertRegionToComputation() local
1759 ConvertToHloModule converter(module, module_builder, true, true, {}, options); in ConvertRegionToComputation()
1777 xla::XlaBuilder module_builder("main"); in ConvertMlirHloToHlo() local
1778 ConvertToHloModule converter(module, module_builder, use_tuple_args, in ConvertMlirHloToHlo()
|