Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetResourceOpInfoForOp (0.29 sec)

  1. 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)
  2. 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