- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for add_attrs (0.33 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
then_branch.set_name(then_branch_xla_func_name); n->ClearAttr("then_branch"); n->AddAttr("then_branch", then_branch); } if (else_branch_has_outside_compilation) { else_branch.set_name(else_branch_xla_func_name); n->ClearAttr("else_branch"); n->AddAttr("else_branch", else_branch); } n->AddAttr(kXlaOriginalOutsideCompilationNodeName, n->name());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
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) -
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) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
// Fetch inputs and outputs from the signature. llvm::SmallVector<mlir::DictionaryAttr, 4> arg_attrs, res_attrs; main_op.getAllArgAttrs(arg_attrs); main_op.getAllResultAttrs(res_attrs); std::vector<std::string> sig_def_inputs = GetStringsFromDictionaryAttr(arg_attrs, kTfSavedModelIndexPathAttr); std::vector<std::string> sig_def_outputs =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/c/c_api.cc
attr_value_proto->length)) { status->status = absl::InvalidArgumentError("Invalid AttrValue proto"); return; } mutex_lock l(graph->mu); op->node.AddAttr(attr_name, attr_val); RecordMutation(graph, *op, "setting attribute"); } void TF_ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, TF_Status* status) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0)