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