Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for clear_attr (0.29 sec)

  1. tensorflow/compiler/jit/encapsulate_util.cc

      if (!s.ok() && s.code() != error::NOT_FOUND) {
        return s;
      }
    
      n->ClearAttr(attr_name);
      attr_value.push_back(value);
      n->AddAttr(attr_name, attr_value);
      return absl::OkStatus();
    }
    
    // Replaces attribute value.
    template <typename T>
    void ReplaceAttr(Node* n, const string& attr_name, const T& value) {
      n->ClearAttr(attr_name);
      n->AddAttr(attr_name, value);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

            (*branch_func.mutable_attr())["_device_ordinal"] = device_ordinal_value;
            n->ClearAttr(attr_name);
            n->AddAttr(attr_name, branch_func);
          }
        } else if (HasNodeAttr(n->def(), "_device_ordinal")) {
          // Function call node containing outside compilation.
          n->ClearAttr("_device_ordinal");
          n->AddAttr("_device_ordinal", device_ordinal_value);
        } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

        dtype = node->attr().at("output_types").list().type(0);
      }
      // Update op name, drop inputs and set attributes required by the Placeholder
      // op.
      *node->mutable_op() = "Placeholder";
      node->clear_attr();
      node->clear_input();
      AddNodeAttr("dtype", dtype, node);
      AddNodeAttr("shape", it->second.shape, node);
      return absl::OkStatus();
    }
    
    // Preprocesses GraphDef before it can be converted to Graph by,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  4. tensorflow/c/python_api.cc

    }
    
    void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
                 TF_Buffer* attr_value_proto, TF_Status* status) {
      TF_SetAttr(graph, op, attr_name, attr_value_proto, status);
    }
    
    void ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
                   TF_Status* status) {
      TF_ClearAttr(graph, op, attr_name, status);
    }
    
    void SetFullType(TF_Graph* graph, TF_Operation* op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jul 12 18:48:56 UTC 2023
    - 3.5K bytes
    - Viewed (0)
  5. tensorflow/c/python_api.h

    void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
                 TF_Buffer* attr_value_proto, TF_Status* status);
    
    // Clears the attr in the node_def Protocol Buffer and sets a status upon
    // completion.
    void ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
                   TF_Status* status);
    
    // Sets the experimental_type` field in the node_def Protocol Buffer.
    void SetFullType(TF_Graph* graph, TF_Operation* op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jul 12 18:48:56 UTC 2023
    - 3.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/xla_cluster_util.cc

    }
    
    void RemoveFromXlaCluster(NodeDef* node_def) {
      node_def->mutable_attr()->erase(kXlaClusterAttr);
    }
    
    void RemoveFromXlaCluster(Node* node) { node->ClearAttr(kXlaClusterAttr); }
    
    namespace {
    typedef xla_config_registry::XlaGlobalJitLevel XlaGlobalJitLevel;
    
    XlaGlobalJitLevel GetXlaGlobalJitLevel(
        const OptimizerOptions::GlobalJitLevel& jit_level_in_session_opts) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/build_xla_ops_pass.cc

        RemoveAllIncomingControlEdges(g, n);
        Operation inverse_predicate_as_control =
            DataToControl(root, inverse_predicated_compilation_key);
        g->AddControlEdge(inverse_predicate_as_control.node(), n);
        n->ClearAttr(kXlaCompiledKernelAttr);
    
        TF_ASSIGN_OR_RETURN(Node* const pco, ReplaceFunctionCallWithPartitionedCall(
                                                 options, flib_def, n, g,
    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. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

        TF_RETURN_IF_ERROR(GetFunctionNameAttr(node, &func_id));
        if (!IsInSubgraph(func_id)) continue;
    
        Subgraph& subgraph = subgraphs_[func_id];
        Node* image = subgraph.MakeNodeImage(graph_in_, node);
        image->ClearAttr(group_attribute_);
        (*node_images)[node] = image;
      }
      return absl::OkStatus();
    }
    
    Status Encapsulator::CopySubgraphEdges(
        const absl::flat_hash_map<const Node*, Node*>& node_images,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  9. tensorflow/c/c_api.cc

    }
    
    void TF_ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
                      TF_Status* status) {
      using tensorflow::RecordMutation;
      mutex_lock l(graph->mu);
      op->node.ClearAttr(attr_name);
      RecordMutation(graph, *op, "clearing attribute");
    }
    
    void TF_SetFullType(TF_Graph* graph, TF_Operation* op,
                        const TF_Buffer* full_type_proto) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top