- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for AddDevicesToOp (0.33 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc
devices.push_back(FakeDevice::Make(gpu0, "compute capability: 7.0")); devices.push_back(FakeDevice::Make(gpu1)); DeviceSet device_set; for (auto& device : devices) device_set.AddDevice(device.get()); AddDevicesToOp(*module_ref, &device_set); auto devices_attr = (*module_ref)->getAttrOfType<mlir::DictionaryAttr>("tf.devices"); ASSERT_NE(devices_attr, nullptr); ASSERT_EQ(devices_attr.size(), 3);
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/tensorflow/utils/device_util.cc
devices->AddGpuDevice(device, gpu_metadata); } 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/mlir_graph_optimization_pass.cc
"fallback enabled."; return absl::OkStatus(); } mlir::OwningOpRef<mlir::ModuleOp> module_ref = std::move(module_ref_status.value()); AddDevicesToOp(*module_ref, &device_set); int per_pass_state_index = 0; bool is_module_updated = false; for (auto& pass_registration : registry_->passes()) { llvm::StringRef name = pass_registration.pass->name();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 18.5K bytes - Viewed (0)