Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for ConstantFoldFallbackHook (0.33 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/constant_fold.h

    #include "mlir/IR/Operation.h"  // from @llvm-project
    #include "mlir/Support/LogicalResult.h"  // from @llvm-project
    
    namespace mlir {
    namespace TF {
    
    LogicalResult ConstantFoldFallbackHook(
        Operation *inst, ArrayRef<Attribute> operands,
        SmallVectorImpl<OpFoldResult> &results);  // NOLINT
    
    }  // namespace TF
    }  // namespace mlir
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Dec 14 20:04:38 UTC 2020
    - 1.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/constant_fold.cc

      return (operands_size <= kOperandsSizeThreshold) &&
             (has_unknown_shape || (results_size <= kResultsSizeThreshold) ||
              (results_size <= kSizeFactor * operands_size));
    }
    
    LogicalResult ConstantFoldFallbackHook(
        Operation* inst, ArrayRef<Attribute> operands,
        SmallVectorImpl<OpFoldResult>& results) {  // NOLINT
      if (!CanBeFolded(inst)) return failure();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.3K bytes
    - Viewed (0)
Back to top