Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for GraphDebugInfo (0.28 sec)

  1. tensorflow/compiler/mlir/python/mlir.cc

    static std::string ImportGraphDefImpl(const std::string& proto,
                                          const std::string& pass_pipeline,
                                          bool show_debug_info,
                                          GraphDebugInfo& debug_info,
                                          GraphImportConfig& specs,
                                          TF_Status* status) {
      GraphDef graphdef;
      auto s = tensorflow::LoadProtoFromBuffer(proto, &graphdef);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:16:49 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc

                             << ", FallbackEnabled: " << num_passes_fallback_enabled
                             << ", Total: " << registry_->passes().size();
      }
    
      GraphDebugInfo debug_info;
      mlir::DialectRegistry registry;
      RegisterDialects(registry);
      mlir::MLIRContext context(registry);
      GraphImportConfig import_config;
      import_config.graph_as_function = true;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 18.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.cc

      TF_RETURN_IF_ERROR(
          tensorflow::LoadProtoFromBuffer({input.data(), input.size()}, &graphdef));
      if (!port::kLittleEndian)
        TF_RETURN_IF_ERROR(ByteSwapTensorContentInGraphDef(&graphdef));
    
      GraphDebugInfo debug_info;
      if (!import_options.debug_info_file.empty()) {
        TF_RETURN_IF_ERROR(
            LoadProtoFromFile(import_options.debug_info_file, &debug_info));
      }
    
      GraphImportConfig specs;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 11:51:44 UTC 2024
    - 14.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h

        llvm::ArrayRef<XlaArgument> args, llvm::ArrayRef<std::string> control_rets,
        llvm::StringRef device_type, const FunctionLibraryDefinition& flib_def,
        const GraphDebugInfo& debug_info,
        llvm::MutableArrayRef<std::unique_ptr<mlir::Pass>>
            custom_legalization_passes = {});
    
    static inline Status CompileToHloGraphAnalysisFailedError() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 10.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/python/converter_python_api.cc

      if (!toco_flags.ParseFromString(toco_flags_proto_txt)) {
        PyErr_SetString(PyExc_ValueError,
                        "Failed to convert Toco to Python String.");
        return nullptr;
      }
    
      tensorflow::GraphDebugInfo debug_info;
      if (debug_info_txt_raw && debug_info_txt_raw != Py_None) {
        std::string debug_info_txt = ConvertArg(debug_info_txt_raw, &error);
        if (error) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

    absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> GraphToModule(
        bool unconditionally_use_set_output_shapes, const Graph& graph,
        llvm::ArrayRef<std::string> control_rets,
        const FunctionLibraryDefinition& flib_def, const GraphDebugInfo& debug_info,
        mlir::MLIRContext* context) {
      mlir::DialectRegistry registry;
      RegisterDialects(registry);
      context->appendDialectRegistry(registry);
      GraphImportConfig config;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_experimental.cc

      if (function_record == nullptr) {
        status->status = tensorflow::errors::NotFound(
            "Unable to find function with name: ", function_name);
        return;
      }
    
      tensorflow::GraphDebugInfo debug_info =
          tensorflow::StackTracesMapToGraphDebugInfo(
              function_record->stack_traces());
    
      string str = debug_info.SerializeAsString();
      void* data = tensorflow::port::Malloc(str.length());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 11 23:52:39 UTC 2024
    - 35.9K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_experimental.h

                                                         TF_Buffer* buf,
                                                         TF_Status* status);
    
    // Get GraphDebugInfo containing stack traces mapping to node names
    TF_CAPI_EXPORT extern void TFE_ContextGetGraphDebugInfo(
        TFE_Context* ctx, const char* function_name, TF_Buffer* buf,
        TF_Status* status);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 39.5K bytes
    - Viewed (0)
Back to top