Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for FusedKernelMatcherPass (0.26 sec)

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

    // Optimizes TF computations by fusing subgraphs/nodes onto more efficient
    // implementations to decrease the number of operations needed to perform a
    // computation.
    struct FusedKernelMatcherPass
        : public impl::FusedKernelMatcherPassBase<FusedKernelMatcherPass> {
      void runOnOperation() override;
    };
    
    bool IsActivationFunction(Operation *op) {
      return isa<EluOp, ReluOp, Relu6Op>(op);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.9K bytes
    - Viewed (0)
Back to top