- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetDeviceOrdinalFromDeviceString (0.32 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc
int64_t device_ordinal0 = -1; mlir::LogicalResult result0 = GetDeviceOrdinalFromDeviceString(unknown_loc, tpu0, &device_ordinal0); EXPECT_TRUE(mlir::succeeded(result0)); EXPECT_EQ(device_ordinal0, 0); int64_t device_ordinal1 = -1; mlir::LogicalResult result1 = GetDeviceOrdinalFromDeviceString(unknown_loc, tpu1, &device_ordinal1); EXPECT_TRUE(mlir::succeeded(result1));
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.h
// Parses a device string and returns its ordinal (id). This will return an // error if the device string is invalid or has no id. mlir::LogicalResult GetDeviceOrdinalFromDeviceString(mlir::Location loc, llvm::StringRef device, int64_t* device_ordinal); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 05 20:02:33 UTC 2020 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
return GetDevicesFromOp(op, dict_attr, devices); } return op->emitOpError( llvm::formatv("unsupported '{0}' attribute", kDevicesAttr)); } mlir::LogicalResult GetDeviceOrdinalFromDeviceString(mlir::Location loc, llvm::StringRef device, int64_t* device_ordinal) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.4K bytes - Viewed (0)