- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for HasResourceInputOrOutput (0.2 sec)
-
tensorflow/compiler/jit/xla_cluster_util.h
// Removes `node` its XLA cluster (by clearing its _XlaCluster attribute). void RemoveFromXlaCluster(Node* node); // Returns true if `node` has a DT_RESOURCE typed input or output. bool HasResourceInputOrOutput(const Node& node); // Determines the global jit level based on GraphOptimizationPassOptions, // --tf_xla_auto_jit and whether the graph is a single GPU graph. OptimizerOptions::GlobalJitLevel GetGlobalJitLevelForGraph(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/jit/partially_decluster_pass.cc
if (IsShapeConsumerOp(*n)) { continue; } // We assume the only XLA-auto-clusterable operations with side effects are // resource variable updates. We can't execute these twice. if (HasResourceInputOrOutput(*n)) { continue; } DeviceType device_type(""); TF_RETURN_IF_ERROR( DeviceNameToDeviceType(n->assigned_device_name(), &device_type));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cluster_util.cc
return std::nullopt; } Status s = AttrValueHasType(*attr_value, "string"); if (!s.ok()) { return std::nullopt; } return attr_value->s(); } bool HasResourceInputOrOutput(const Node& node) { return std::find(node.input_types().begin(), node.input_types().end(), DT_RESOURCE) != node.input_types().end() ||
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/mark_for_compilation_pass.cc
: node->requested_device(); TF_ASSIGN_OR_RETURN(DeviceId device, device_info_cache_.GetIdFor(device_name_str)); bool is_resource_op = HasResourceInputOrOutput(*node); std::optional<DeviceId> resource_op_device; if (is_resource_op) { resource_op_device = device; } std::optional<int> resource_var_operation_node_id;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0)