- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for IsOnGpuDevice (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/rewrite_util.td
// 32 bit integers are supported for the values. class IsConstantValueOf<int value> : Constraint<CPred< "IsConstantValueOf($0," # value # ")">>; def IsOnGpuDevicePred : CPred<"IsOnGpuDevice($0.getOwner())">;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 21 02:54:35 UTC 2022 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/rewrite_util.cc
return std::nullopt; } if (!parsed_name.has_type) { return std::nullopt; } return parsed_name.type; } } // namespace bool IsOnGpuDevice(mlir::Operation *op) { std::optional<std::string> device = GetOpDevice(op); if (!device) return false; return *device == kDeviceGpu; } void CopyDeviceAndUnderscoredAttributesAdaptor(mlir::OpResult src,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/rewrite_util.h
return true; } return false; } // Returns true if `op` is placed on GPU device, and false if it's on other // devices or the device is not specified. bool IsOnGpuDevice(mlir::Operation *op); // Wrappers for CopyDeviceAndUnderscoredAttributes void CopyDeviceAndUnderscoredAttributesAdaptor(mlir::OpResult src, mlir::OpResult dest);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 4K bytes - Viewed (0)