Method FusionCount

tensorflow/compiler/xla/service/defuser_test.cc:31–39

Returns the number of fusion instructions in the module.

 protected:
  // Returns the number of fusion instructions in the module.
  int FusionCount(const HloModule* m) {
    int count = 0;
    for (HloComputation* computation : m->computations()) {
      if (computation->IsFusionComputation()) {
        count++;
      }
    }
    return count;
  }

  Defuser defuser_;
  const Shape shape_ = ShapeUtil::MakeShape(F32, {2, 2});