- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetResourceOpInfoForOp (0.17 sec)
-
tensorflow/compiler/jit/compilability_check_util.h
bool IsStackOp(const Node& node) const { const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(node.type_string()); return op_info && op_info->resource_kind() == XlaResourceKind::kStack; } bool IsTensorArrayOp(const Node& node) const { const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(node.type_string()); return op_info && op_info->resource_kind() == XlaResourceKind::kTensorArray;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/jit/resource_operation_safety_analysis.cc
} if (should_ignore) { *out_resource_op_kind = std::nullopt; return absl::OkStatus(); } const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(n.type_string()); if (op_info) { *out_resource_op_kind = op_info->kind(); return absl::OkStatus(); } // We conservatively assume that functions will both read and write resource
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
// fold, say, a TensorArrayRead or a StackPopV2). But we don't // auto-cluster these operations today so we're good for now. const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(node->type_string()); bool is_tensor_array_or_stack_op = op_info && op_info->resource_kind() != XlaResourceKind::kVariable; if (!is_tensor_array_or_stack_op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0)