Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 12 for output_arrays (0.15 sec)

  1. tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/output_arrays.mlir

    Tres Popp <******@****.***> 1648106941 -0700
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 24 07:35:24 UTC 2022
    - 1.1K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. tensorflow/compiler/mlir/lite/python/graphdef_to_tfl_flatbuffer.cc

          ParseInputArrayInfo(node_names, node_dtypes, node_shapes, &specs.inputs));
    
      // Parse output arrays.
      std::vector<std::string> output_arrays(model_flags.output_arrays().begin(),
                                             model_flags.output_arrays().end());
      TF_RETURN_IF_ERROR(ParseOutputArrayInfo(output_arrays, &specs.outputs));
    
      // Parse control output arrays.
      std::vector<std::string> control_output_arrays(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 11 19:29:56 UTC 2024
    - 5.1K bytes
    - Viewed (0)
  5. 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)
  6. 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)
  7. 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)
  8. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_cl.h

    // Import options.
    extern llvm::cl::opt<std::string> input_arrays;
    extern llvm::cl::opt<std::string> input_dtypes;
    extern llvm::cl::opt<std::string> input_shapes;
    extern llvm::cl::opt<std::string> output_arrays;
    extern llvm::cl::opt<std::string> control_output_arrays;
    extern llvm::cl::opt<std::string> inference_type;
    extern llvm::cl::opt<std::string> min_values;
    extern llvm::cl::opt<std::string> max_values;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 10 20:59:50 UTC 2023
    - 2.3K bytes
    - Viewed (0)
  9. 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)
  10. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_cl.cc

            "Input tensor shapes. Shapes for different tensors are separated by "
            "':', and dimension sizes for the same tensor are separated by ','"),
        llvm::cl::init(""));
    
    // NOLINTNEXTLINE
    opt<std::string> output_arrays(
        "tf-output-arrays", llvm::cl::desc("Output tensor names, separated by ','"),
        llvm::cl::init(""));
    
    // NOLINTNEXTLINE
    opt<std::string> control_output_arrays(
        "tf-control-output-arrays",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 10 20:59:50 UTC 2023
    - 5.5K bytes
    - Viewed (0)
Back to top