Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 23 of 23 for ParseFullName (0.26 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc

      if (!iter) return false;
      auto device = op->getAttrOfType<StringAttr>(kDeviceAttr);
      if (!device) return false;
      tensorflow::DeviceNameUtils::ParsedName parsed_device;
      if (!tensorflow::DeviceNameUtils::ParseFullName(device.getValue().str(),
                                                      &parsed_device)) {
        return false;
      }
      return parsed_device.type == "CPU";
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 29.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc

                                  llvm::SmallVectorImpl<Device>* parsed_devices) {
      parsed_devices->reserve(device_names.size());
      for (const auto& device_name : device_names) {
        Device parsed_name;
        if (!DeviceNameUtils::ParseFullName(device_name, &parsed_name))
          return false;
    
        parsed_devices->push_back(parsed_name);
      }
      return true;
    }
    
    using DeviceNames = llvm::SmallVector<std::string, 8>;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 46.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

      // it. Note that in current TF, placer will place full device name to each
      // node.
      DeviceNameUtils::ParsedName parsed_name;
      if (!DeviceNameUtils::ParseFullName(node_def.device(), &parsed_name)) {
        return errors::InvalidArgument(
            "Op ", op_name, " has invalid device name: ", node_def.device());
      }
      // Keep the parsed name untouched if the device name is empty.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top