Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for IdentityArithmeticOpFolder (0.47 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_arith_ops_folder.h

    // known identity value is either lhs or rhs.
    template <
        typename OpT,
        typename std::enable_if<llvm::is_one_of<
            OpT, AddV2Op, SubOp, MulOp, DivOp, RealDivOp>::value>::type * = nullptr>
    OpFoldResult IdentityArithmeticOpFolder(OpT arithmetic_op,
                                            ArrayRef<Attribute> operands) {
      auto lhs_type = mlir::cast<ShapedType>(arithmetic_op.getX().getType());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      results.add<AddV2OfNegLeft, AddV2OfNegRight>(context);
    }
    
    OpFoldResult AddV2Op::fold(FoldAdaptor adaptor) {
      auto operands = adaptor.getOperands();
      return IdentityArithmeticOpFolder<AddV2Op>(*this, operands);
    }
    
    //===----------------------------------------------------------------------===//
    // AllOp
    //===----------------------------------------------------------------------===//
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      results.add<RealDivWithSqrtDivisor, RealDivWithConstDivisor>(context);
    }
    
    OpFoldResult RealDivOp::fold(FoldAdaptor adaptor) {
      auto operands = adaptor.getOperands();
      return IdentityArithmeticOpFolder<RealDivOp>(*this, operands);
    }
    
    //===----------------------------------------------------------------------===//
    // ReluOp
    //===----------------------------------------------------------------------===//
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top