Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 43 for add_attrs (0.12 sec)

  1. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

          });
    
      for (auto& src_arg : src_arg_pairs) {
        if (guaranteed_const_nodes.count(src_arg.first) != 0) {
          VLOG(1) << "Guaranteed const found: " << src_arg.first->DebugString();
          src_arg.second->AddAttr("_is_guaranteed_constant", true);
        }
      }
    }
    
    struct OutputInputTensorPairHasher {
      uint64 operator()(std::pair<OutputTensor, InputTensor> const& s) const {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

                                    {}, {{{"n_c"}, "UncompilableUnary", {"n_a"}}});
      FunctionDef noinline = compilable;
      noinline.mutable_signature()->set_name("NoInlineFn");
      AddAttr("_noinline", static_cast<bool>(true), noinline.mutable_attr());
    
      FunctionDefLibrary flib;
      *flib.add_function() = compilable;
      *flib.add_function() = uncompilable;
      *flib.add_function() = noinline;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc

              .node();
    
      TF_RETURN_IF_ERROR(main_scope.status());
    
      std::vector<string> compile_time_const_inputs;
      compile_time_const_inputs.push_back("size");
      (*result)->AddAttr(kXlaCompileTimeConstantInputsAttr,
                         compile_time_const_inputs);
      return status;
    }
    
    void ReplaceTensorFlowSliceWithStaticShapedSlice(Graph* g, Node* slice,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 13.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/compilability_check_util_test.cc

      auto opts = builder.opts();
      Node* const0 = ops::SourceOp("InputFloatOp", opts);
      Node* uncompilable_op = ops::UnaryOp("MissingKernel", const0, opts);
      uncompilable_op->AddAttr("_xla_outside_compilation", "0");
      GraphDef graph_def;
      TF_EXPECT_OK(builder.ToGraphDef(&graph_def));
    
      auto* flib_runtime = GetFunctionLibraryRuntime();
    
      // Outside compiled ops are considered by default..
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 10 12:32:39 UTC 2022
    - 22.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

        const absl::InlinedVector<Node*, 4>& control_ret_nodes) {
      // Store the arg/return attributes as a list rather than uniqueuing during
      // construction.
      llvm::SmallVector<mlir::NamedAttrList, 4> arg_attrs;
      arg_attrs.resize(func.getNumArguments());
      llvm::SmallVector<mlir::NamedAttrList, 4> ret_attrs;
      ret_attrs.resize(func.getNumResults());
    
      auto set_attributes_on_func = [&](Node* node, int64_t index, bool is_arg) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

      auto x = ops::Placeholder(root.WithOpName("x"), DT_FLOAT);
      auto add1 = ops::Add(root.WithOpName("add1"), x, x);
      add1.node()->AddAttr("_cluster", "cluster1");
      auto add2 = ops::Add(root.WithOpName("add2"), add1, add1);
      add2.node()->AddAttr("_cluster", "cluster2");
      auto out = ops::Mul(root.WithOpName("mul"), add1, add2);
    
      Graph graph_before_encapsulation(OpRegistry::Global());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/build_xla_ops_pass.cc

                                   cluster_info.function);
    
      bool has_ref_attr;
      TF_RETURN_IF_ERROR(
          GetNodeAttr(n->attrs(), kXlaHasReferenceVarsAttr, &has_ref_attr));
      xla_compile.operation.node()->AddAttr(kXlaHasReferenceVarsAttr, has_ref_attr);
      TF_RETURN_IF_ERROR(
          CopyIncomingControlEdges(g, /*from=*/n, /*to=*/xla_compile.key.node()));
    
      std::vector<Output> xla_run_args =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  8. api/go1.21.txt

    pkg log/slog, method (*Logger) Warn(string, ...interface{}) #56345
    pkg log/slog, method (*Logger) WithGroup(string) *Logger #56345
    pkg log/slog, method (*Logger) With(...interface{}) *Logger #56345
    pkg log/slog, method (*Record) AddAttrs(...Attr) #56345
    pkg log/slog, method (*Record) Add(...interface{}) #56345
    pkg log/slog, method (Record) Attrs(func(Attr) bool) #59060
    pkg log/slog, method (Record) Clone() Record #56345
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Mon Aug 07 09:39:17 UTC 2023
    - 25.6K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api.cc

    const TFE_OpAttrs* TFE_OpGetAttrs(const TFE_Op* op) {
      return tensorflow::wrap(tensorflow::unwrap(op)->GetOpAttrs());
    }
    
    void TFE_OpAddAttrs(TFE_Op* op, const TFE_OpAttrs* attrs) {
      tensorflow::unwrap(op)->AddAttrs(tensorflow::unwrap(attrs));
    }
    
    void TFE_OpAttrsSerialize(const TFE_OpAttrs* attrs, TF_Buffer* buf,
                              TF_Status* status) {
      tensorflow::NameAttrList name_and_attrs;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/mark_for_compilation_pass.cc

            }
            absl::StrAppend(&name,
                            GetNextClusterSequenceNumber(graph_fingerprint_));
          }
    
          n->AddAttr(kXlaClusterAttr, name);
          n->AddAttr(kXlaAlreadyClustered, true);
          VLOG(3) << "Assigning node " << n->name() << " to cluster " << name;
        }
      }
    
      return absl::OkStatus();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top