Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for IsDeviceClusterOp (0.12 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/verify_no_outside_compilation_markers_pass.cc

              VerifyNoOutsideCompilationMarkersPass> {
     public:
      void runOnOperation() override;
    };
    
    bool IsLaunchOp(Operation& op) {
      return dyn_cast<tf_device::LaunchOp>(op) != nullptr;
    }
    
    bool IsDeviceClusterOp(Operation& op) {
      return dyn_cast<tf_device::ClusterOp>(op) != nullptr;
    }
    
    bool HasChildLaunchDeviceOp(Operation& op) {
      auto cluster_op = dyn_cast<tf_device::ClusterOp>(op);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Aug 22 19:52:08 UTC 2023
    - 3.5K bytes
    - Viewed (0)
Back to top