- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for clear_attr (0.29 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)