Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 22 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/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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/shape-inference.mlir

        func.return %0 : tensor<?x19xf32>
      }
    }
    
    // CHECK-LABEL: HloModule main
    // CHECK:       (arg_tuple.{{[0-9]+}}: (f32[10,17], f32[17,19])) -> (f32[10,19])
    
    // NO_TUPLES-LABEL: HloModule main
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 23 18:56:13 UTC 2022
    - 969 bytes
    - Viewed (0)
Back to top