Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for VerifyMulOpShapeConstraints (0.23 sec)

  1. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

            /*max_bcast_rank=*/6);
      }
      return false;
    }
    
    // Return true if the given Mul operation has the CPU kernel supported shapes.
    bool VerifyMulOpShapeConstraints(MulOp op) {
      auto element_type = getElementTypeOrSelf(op.getOutput().getType());
    
      // Allows QI8 and QUI8 inputs up to five dimension broadcasting unless the
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        Pure,
        Commutative,
        QuantizableResult,
        BinaryOpSameElementTypeConstraint,
        TFL_RuntimePredOpTrait<"Operands do not have valid shapes",
          CPred<"TFL::VerifyMulOpShapeConstraints(llvm::cast<MulOp>($_op))">>,
        DeclareOpInterfaceMethods<TFL_ArithmeticCount>]> {
      let summary = "Multiplication operator";
    
      let description = [{
        Element-wise multiplication operation.
      }];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top