- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 22 for hlo_module (0.15 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/tf2xla/api/v2/legalize_tf.cc
return absl::OkStatus(); } TF_ASSIGN_OR_RETURN( auto hlo_module_config, xla::HloModule::CreateModuleConfigFromProto( compilation_result->computation->proto(), xla::DebugOptions())); TF_ASSIGN_OR_RETURN( std::unique_ptr<xla::HloModule> hlo_module, xla::HloModule::CreateFromProto(compilation_result->computation->proto(), hlo_module_config));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 00:40:46 UTC 2024 - 6.8K 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/jit/xla_compilation_cache.proto
// serialize this rather than the input graphdef because it provides a // stronger guarantee over what bindings are needed between the HLO and // calling TF graph. xla.HloModuleProto hlo_module = 2; // The raw bytes of the executable. bytes executable = 3;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/jax_to_tfl_flatbuffer.cc
LOG(ERROR) << "HLO Module loading failed: " << hlo_module_error.status(); return nullptr; } auto hlo_module = std::move(hlo_module_error.value()); mlir::OwningOpRef<mlir::ModuleOp> module = mlir::ModuleOp::create(mlir::UnknownLoc::get(context)); auto status = ConvertHloToMlirHlo(*module, hlo_module.get(), import_all_computations); if (!status.ok()) { LOG(ERROR) << "HLO Module import failed: " << status;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/device_compiler_test_helper.cc
auto file_path = io::JoinPath(persistent_cache_dir_path, file_name); TF_RETURN_IF_ERROR(ReadTextOrBinaryProto(env, file_path, &entry)); entry.mutable_hlo_module()->set_name( absl::StrCat(entry.hlo_module().name(), "_altered")); TF_RETURN_IF_ERROR(WriteBinaryProto(env, file_path, entry)); altered = true; } } if (!altered) { return errors::NotFound(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 6.2K 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/mlir/tensorflow/tests/compile_mlir_util/add.mlir
func.func @main(%arg0: tensor<f32>, %arg1: tensor<f32>) -> tensor<f32> { %0 = "tf.AddV2"(%arg0, %arg1) : (tensor<f32>, tensor<f32>) -> tensor<f32> func.return %0 : tensor<f32> } } // CHECK-LABEL: HloModule main // CHECK: ENTRY %main.{{[0-9]+}} ([[ARG0:.*]]: f32[], [[ARG1:.*]]: f32[]) -> (f32[]) { // CHECK-NEXT: %[[ARG0]] = f32[] parameter(0) // CHECK-NEXT: %[[ARG1]] = f32[] parameter(1)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 23 18:56:13 UTC 2022 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/shape-inference.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 23 18:56:13 UTC 2022 - 969 bytes - Viewed (0)