Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for hlo_module (0.12 sec)

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