Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for argv_index (0.15 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

      for (const auto& input_pair : llvm::enumerate(signature->inputs)) {
        const int arg_index = GetTensorIndex(
            tensors[input_pair.value()->tensor_index]->name, input_names);
        if (arg_index == -1) {
          func->emitWarning("Invalid signature tensors specified.");
          return;
        }
        func.setArgAttr(
            arg_index, kTfSavedModelIndexPathAttr,
            mlir::ArrayAttr::get(context, {mlir::StringAttr::get(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

      int arg_index = iter->second;
      if (inserted) {
        NodeDef arg_def;
        NodeDefBuilder builder(
            absl::StrCat(src_node->name(), "_", src_slot, "_arg"), kArgOp,
            NodeDebugInfo(src_node->def()));
        DataType dtype = edge->dst()->input_type(edge->dst_input());
        builder.Attr("T", dtype);
        builder.Attr("index", arg_index);
        Status s = builder.Finalize(&arg_def);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
Back to top