Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for IsResourceMergeable (0.21 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc

     * are typically associated with packed variables. Presently, we assume this
     * set spans all the devices. So, a variable on a `COMPOSITE` will have a local
     * instance on the execute op's device.
     */
    bool IsResourceMergeable(Attribute& resource_attr, Attribute& device_attr) {
      return resource_attr &&
             ((resource_attr == device_attr) ||
              (resource_attr.cast<mlir::StringAttr>().getValue().find(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 17:52:11 UTC 2024
    - 27K bytes
    - Viewed (0)
Back to top