Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 17 for c_input_names (0.28 sec)

  1. tensorflow/c/c_api.cc

      for (int i = 0; i < ninputs; ++i) {
        input_names[i] = c_input_names[i];
      }
      for (int i = 0; i < noutputs; ++i) {
        output_names[i] = c_output_names[i];
      }
      for (int i = 0; i < ntargets; ++i) {
        target_oper_names[i] = c_target_oper_names[i];
      }
      string new_handle;
      status->status = s->session->PRunSetup(input_names, output_names,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc

            const std::string inputs_attr_str =
                mlir::cast<StringAttr>(inputs_attr).getValue().str();
            std::vector<std::string> fn_input_names =
                absl::StrSplit(inputs_attr_str, ',', absl::SkipEmpty());
    
            for (StringRef input_name : fn_input_names) {
              if (input_name_set.contains(input_name)) {
                // Found a duplicated name, all input names will be prefixed by
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/saved_model/saved_model.cc

          }
        }
    
        for (auto func_name : func_names) {
          TFRTSavedModelSignatureInfo sig_info;
          sig_info.func_name = func_name;
          sig_info.input_names = input_names;
          sig_info.input_specs = input_specs;
          sig_info.input_devices = input_devices;
          sig_info.output_names = output_names;
          sig_info.output_specs = output_specs;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/python/jax_to_tfl_flatbuffer.cc

      // Retrieve input names from model flags.
      std::vector<std::string> input_names;
      for (const auto& input : model_flags.input_arrays()) {
        input_names.push_back(input.name());
      }
    
      const auto& inputs = absl::StrJoin(input_names, ",");
      mlir::OpBuilder builder(*module);
      llvm::SmallVector<mlir::NamedAttribute> attrs;
      attrs.push_back(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 11 19:29:56 UTC 2024
    - 8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc

    }  // namespace
    
    FailureOr<SmallVector<AssetFileDef>> ConvertAssetArgs(ModuleOp module_op) {
      func::FuncOp main_func_op = FindMainFuncOp(module_op);
      if (!main_func_op) return failure();
    
      SmallVector<StringRef> input_names = GetEntryFunctionInputs(main_func_op);
      SymbolTable symbol_table(module_op);
      SmallVector<AssetFileDef> asset_file_defs;
    
      for (BlockArgument argument : main_func_op.getArguments()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfrt/saved_model/saved_model.h

    // be used to retrieve these information in a tf_saved_model module.
    struct TFRTSavedModelSignatureInfo {
      llvm::StringRef func_name;
    
      // The following are metadata for inputs.
      llvm::ArrayRef<llvm::StringRef> input_names;
      llvm::ArrayRef<
          std::pair<tensorflow::DataType, tensorflow::PartialTensorShape>>
          input_specs;
      llvm::ArrayRef<llvm::StringRef> input_devices;
    
      // The following are metadata for outputs.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jan 27 17:42:41 UTC 2024
    - 2.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

              exporter.op_to_name_.GetUniqueName(tensor_id_node));
        }
      }
    
      if (!input_names.empty()) {
        TF_RET_CHECK(input_names.size() == block.getNumArguments());
        for (const auto& it : llvm::enumerate(function.getArguments())) {
          // TODO(lyandy): Update when changing feed/fetch import.
          std::string name(input_names[it.index()]);
          assert(!name.empty() && "expected non-empty name");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

              exporter.op_to_name_.GetUniqueName(tensor_id_node));
        }
      }
    
      if (!input_names.empty()) {
        TF_RET_CHECK(input_names.size() == block.getNumArguments());
        for (const auto& it : llvm::enumerate(function.getArguments())) {
          // TODO(lyandy): Update when changing feed/fetch import.
          std::string name(input_names[it.index()]);
          assert(!name.empty() && "expected non-empty name");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/python/saved_model_to_tfl_flatbuffer.cc

      auto input_attr = tf_attrs.get("inputs");
      if (!input_attr) {
        return errors::InvalidArgument("no inputs attribute found");
      }
      auto input_names = mlir::cast<mlir::StringAttr>(input_attr).getValue();
      input_names.split(function_input_names, ",", /*MaxSplit=*/-1,
                        /*KeepEmpty=*/false);
      const int function_input_names_size = function_input_names.size();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun May 12 12:39:37 UTC 2024
    - 11K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/python/mlir.h

    std::string ImportGraphDef(const std::string &proto,
                               const std::string &pass_pipeline,
                               bool show_debug_info, absl::string_view(input_names),
                               absl::string_view(input_data_types),
                               absl::string_view(input_data_shapes),
                               absl::string_view(output_names), TF_Status *status);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 14 23:44:01 UTC 2023
    - 5.6K bytes
    - Viewed (0)
Back to top