- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for hlo_module (0.12 sec)
-
tensorflow/compiler/jit/device_executable_persistor.h
// match exactly. if (!disable_strict_signature_checks_) { if (!AreSerializedProtosEqual(hlo_module, entry.hlo_module())) { VLOG(2) << "HLOs do not match:\n" << "got:\n" << hlo_module.DebugString() << "\nexpected:\n" << entry.hlo_module().DebugString() << "\n"; return errors::InvalidArgument("Serialized HLO does not match."); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
auto status_or_hlo_module = xla::HloModule::CreateFromProto( compilation_result.computation->proto(), module_config); if (!status_or_hlo_module.ok()) { LOG(ERROR) << "Conversion to HLO module failed: " << status_or_hlo_module.status(); return mlir::failure(); } xla::HloModule* hlo_module = status_or_hlo_module.value().get(); output << hlo_module->ToString();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
TF_ASSIGN_OR_RETURN(auto hlo_module_config, xla::HloModule::CreateModuleConfigFromProto( computation.proto(), debug_options)); TF_ASSIGN_OR_RETURN( std::unique_ptr<xla::HloModule> hlo_module, xla::HloModule::CreateFromProto(computation.proto(), hlo_module_config)); if (!RootInstructionIsTuple(*hlo_module)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
if (lower_to_xla_hlo) { TF_RETURN_IF_ERROR(mlir::ConvertMlirHloToHlo( module_op, &hlo_proto, use_tuple_args, return_tuple, options)); *xla_computation = xla::XlaComputation(hlo_proto.hlo_module()); } return absl::OkStatus(); } // Wraps the optional lowering version to keep the api the same for clients. Status ConvertMLIRToXlaComputation(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
result.computation->GetProgramShape()); xla::HloModuleConfig config(program_shape); TF_ASSIGN_OR_RETURN( std::unique_ptr<xla::HloModule> new_module, xla::HloModule::CreateFromProto(result.computation->proto(), config)); xla::HloPrintOptions opts; if (stage == IrExportStage::HLO_NO_METADATA) { opts.set_print_metadata(false); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0)