Searched refs:module_proto (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_module_importer.cc | 56 Status HloModuleImporter::Import(const xla::HloModuleProto& module_proto) { in Import() argument
60 xla::HloModule::CreateModuleConfigFromProto(module_proto, debug_options)); in Import()
62 module_proto, module_config)); in Import()
|
| D | xla_mlir_translate.cc | 125 const HloModuleProto& module_proto = hlo_proto.hlo_module(); in HloModuleFromProto() local
128 module_proto, GetDebugOptionsFromFlags())); in HloModuleFromProto()
129 return HloModule::CreateFromProto(module_proto, module_config); in HloModuleFromProto()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner_interface.cc | 77 HloModuleProto module_proto; in ReadModuleFromModuleBinaryProtofile() local
79 filename, &module_proto)); in ReadModuleFromModuleBinaryProtofile()
83 HloModule::CreateModuleConfigFromProto(module_proto, debug_options)); in ReadModuleFromModuleBinaryProtofile()
85 return HloModule::CreateFromProto(module_proto, module_config); in ReadModuleFromModuleBinaryProtofile()
|
| D | hlo_module_group.cc | 88 const HloModuleProto& module_proto = proto.hlo_modules(i); in CreateFromProto() local
91 HloModule::CreateFromProto(module_proto, module_configs[i])); in CreateFromProto()
|
| D | service.cc | 757 const HloModuleProto& module_proto, in BuildExecutable() argument
763 module_proto.name()); in BuildExecutable()
767 CreateModuleFromProto(module_proto, *module_config, run_backend_only)); in BuildExecutable()
780 if (DumpingEnabledForHloModule(module_proto.name(), debug_opts) && in BuildExecutable()
783 *hlo_proto->mutable_hlo_module() = module_proto; in BuildExecutable()
|
| D | service.h | 236 const HloModuleProto& module_proto,
|
| D | dump.cc | 240 HloProto module_proto = in DumpHloModuleImpl() local
243 if (!tensorflow::SerializeToStringDeterministic(module_proto, &pb)) { in DumpHloModuleImpl()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_conversions.cc | 389 xla::HloModuleProto module_proto = in FromC() local
393 module_proto, ApiConverter::FromC(c_module.module_config)); in FromC()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 92 const HloModuleProto& module_proto = hlo_proto.hlo_module(); in HloModuleFromProto() local
95 module_proto, xla::GetDebugOptionsFromFlags())); in HloModuleFromProto()
96 return HloModule::CreateFromProto(module_proto, module_config); in HloModuleFromProto()
|