Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for HasResourceInputOrOutput (0.2 sec)

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