Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetResourceOpInfoForOp (0.17 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top