- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for mutable_hlo_module (0.48 sec)
-
tensorflow/compiler/mlir/lite/python/jax_to_tfl_flatbuffer.cc
parser.RecordErrorsTo(&collector); return hlo_proto->ParseFromString(contents) || parser.ParseFromString(contents, hlo_proto) || hlo_proto->mutable_hlo_module()->ParseFromString(contents) || parser.ParseFromString(contents, hlo_proto->mutable_hlo_module()); } mlir::OwningOpRef<mlir::ModuleOp> HloToMlirHloTranslateFunction( mlir::StringRef input, mlir::MLIRContext* context, bool import_all_computations) {
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
XlaSerializedCacheEntry entry; 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) {
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/jit/device_executable_persistor.h
const xla::HloModuleProto& hlo_module = compilation_result.computation->proto(); *serialized_entry.mutable_key() = BuildSerializedCacheKey(signature_hash, hlo_module); *serialized_entry.mutable_hlo_module() = hlo_module; // XLA compiler supports exporting executables as an AOT compilation result // to avoid running potentially expensive compilation pipeline twice.
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
xla::XlaComputation computation, return_value.valid() ? builder.Build(return_value) : builder.Build()); auto hlo_module = computation.proto(); xla::HloProto hlo_proto; hlo_proto.mutable_hlo_module()->Swap(&hlo_module); compilation_result->computation = std::make_shared<xla::XlaComputation>(); xla::XlaComputation* xla_computation = compilation_result->computation.get();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0)