- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetNodeAttr (0.09 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
recv_at_host_dtypes->resize(arg_nodes.size(), DT_INVALID); for (auto* n : arg_nodes) { int index; TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); DataType dtype; TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "T", &dtype)); (*recv_at_host_dtypes)[index] = dtype; } for (int i = 0, end = recv_at_host_dtypes->size(); i < end; i++) {
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/c/c_api.cc
cpp_type v; \ status->status = \ tensorflow::GetNodeAttr(oper->node.attrs(), attr_name, &v); \ if (!status->status.ok()) return; \ *value = static_cast<c_type>(v); \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
for (const Node* node : graph.nodes()) { if (node->name() == name) return node; } return nullptr; } bool HasGuaranteeConstAttr(const Node& n) { bool is_guaranteed_constant = false; if (!GetNodeAttr(n.attrs(), "_is_guaranteed_constant", &is_guaranteed_constant) .ok()) { return false; } return is_guaranteed_constant; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0)