Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for mutable_hlo_module (0.14 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
Back to top