- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TFDeviceAssignmentByFuncAttrPass (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_assignment.cc
// If device attribute is not found from the function, nothing is done. class TFDeviceAssignmentByFuncAttrPass : public impl::TFDeviceAssignmentByFuncAttrPassBase< TFDeviceAssignmentByFuncAttrPass> { public: TFDeviceAssignmentByFuncAttrPass() = default; TFDeviceAssignmentByFuncAttrPass(const TFDeviceAssignmentByFuncAttrPass&) {} void runOnOperation() override { func::FuncOp func = getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
%2 = "tf.Const"() {device = "baz", value = dense<[[42.0]]> : tensor<1x1xf32>} : () -> tensor<1x1xf32> ``` }]; } def TFDeviceAssignmentByFuncAttrPass : Pass<"tf-device-assignment-by-func-attr", "mlir::func::FuncOp"> { let summary = "Device assignment in TF dialect using the device specified in the function attribute.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)