Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for output_arrays (0.18 sec)

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

      for (const auto& output_array : model_flags.output_arrays()) {
        if (function_output_names_set.count(output_array) == 0) {
          return errors::InvalidArgument("output array name (", output_array,
                                         ") does not exist in the given graph");
        }
      }
      return absl::OkStatus();
    }
    
    Status ConvertSavedModelToTFLiteFlatBuffer(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun May 12 12:39:37 UTC 2024
    - 11K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.cc

        const std::vector<std::string>& output_arrays,
        const std::vector<std::string>& control_output_arrays,
        const GraphdefToMlirOptions& import_options, mlir::MLIRContext* context) {
      auto module_or = GraphdefToMlirImport(
          input, input_arrays, input_dtypes, input_shapes, output_arrays,
          control_output_arrays, import_options, context);
      if (!module_or.status().ok()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 11:51:44 UTC 2024
    - 14.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.h

        llvm::StringRef input, const std::vector<std::string>& input_arrays,
        const std::vector<std::string>& input_dtypes,
        const std::vector<std::optional<std::vector<int>>>& input_shapes,
        const std::vector<std::string>& output_arrays,
        const std::vector<std::string>& control_output_arrays,
        const GraphdefToMlirOptions& import_options, mlir::MLIRContext* context);
    
    ABSL_DEPRECATED(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 5.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.h

    // source into a MLIR module. If `input_mlir` is true, load from a MLIR source
    // file; otherwise, load from a GraphDef.
    // Setting prune_unused_nodes to true, would prune unreachable nodes if
    // output_arrays is specified.
    absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> LoadFromGraphdefOrMlirSource(
        const std::string& input_filename, bool input_mlir,
        bool use_splatted_constant, const std::vector<std::string>& extra_tf_opdefs,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 08:30:24 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc

          enable_soft_placement,  set_original_tf_func_name};
    
      auto module_or = tensorflow::GraphdefToMlirTranslateFunction(
          input, input_arrays, input_dtypes, input_shapes, output_arrays,
          control_output_arrays, options, context);
      if (!module_or.status().ok()) return nullptr;
      return std::move(module_or).value();
    }
    
    static TranslateToMLIRRegistration GraphdefToMlirTranslate(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc

            file->getBuffer(), input_arrays, input_dtypes, input_shapes,
            output_arrays, control_output_arrays, graphdef_conversion_options,
            context);
      }
      return GraphdefToMlirTranslateFunction(file->getBuffer(), input_arrays,
                                             input_dtypes, input_shapes,
                                             output_arrays, control_output_arrays,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/tf_tfl_translate.cc

        module = tensorflow::LoadFromGraphdefOrMlirSource(
            input_file_name, input_mlir, use_splatted_constant, custom_opdefs,
            specs, debug_info_file, input_arrays, input_dtypes, input_shapes,
            output_arrays, control_output_arrays, &source_mgr, &context);
      }
    
      // If errors occur, the library call in the above already logged the error
      // message. So we can just return here.
      if (!module.ok()) return kTrFailure;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 14K bytes
    - Viewed (0)
Back to top