- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for DeviceSet (0.14 sec)
-
tensorflow/compiler/jit/device_util.h
explicit DeviceId(int id) : id_(id) {} int id() const { return id_; } friend class DeviceInfoCache; friend class DeviceSet; }; // A set of DeviceIds, represented as a bitmap. class DeviceSet { public: void Insert(DeviceId device_id); void UnionWith(const DeviceSet& other); bool IsEmpty() const; // Calls `func` on each DeviceId in the set. Stops iterating early if `func` // return false.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 17:18:31 UTC 2024 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
} else { devices->AddDevice(device); } } return mlir::success(); } } // namespace void AddDevicesToOp(mlir::Operation* op, const DeviceSet* device_set) { if (!device_set) return; mlir::MLIRContext* ctx = op->getContext(); mlir::Builder builder(ctx); // Collect devices with attached metadata.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc
#include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/platform/test.h" #include "tensorflow/core/util/device_name_utils.h" namespace tensorflow { namespace { // A fake device used to populate a DeviceSet. class FakeDevice : public Device { public: explicit FakeDevice(const DeviceAttributes& device_attributes) : Device(nullptr, device_attributes) {}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc
mlir::func::registerAllExtensions(registry); // clang-format on } Status MlirFunctionOptimizationPass::Run( const std::string& function_name, const DeviceSet& device_set, const ConfigProto& config_proto, const FunctionOptimizationPass::FunctionOptions& function_options, std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 18.5K bytes - Viewed (0)