Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for resource_attr (0.22 sec)

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

     * 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(
                   "COMPOSITE") != llvm::StringRef::npos));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 17:52:11 UTC 2024
    - 27K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/saved_model/core/revived_types/partially_revived_objects.cc

          }
          *handle = assets_iter->second->handle();
          return Status();
        }
        case SavedObject::kResource: {
          const auto& resource_iter = objects.restored_resources.find(node_id);
          if (resource_iter == objects.restored_resources.end()) {
            return absl::FailedPreconditionError(absl::StrCat(
                "Tried to convert node id ", node_id,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 09 20:11:48 UTC 2023
    - 23.7K bytes
    - Viewed (0)
Back to top