- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for VerifyAddOpShapeConstraints (0.31 sec)
-
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
op.getBiasMutable().assign(none_value); return success(); } return failure(); } }; // Return true if the given Add operation has the CPU kernel supported shapes. bool VerifyAddOpShapeConstraints(AddOp op) { auto element_type = getElementTypeOrSelf(op.getOutput().getType()); // Allows F32, QI8, QUI8 and I32 outputs when the operands have valid shapes,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
} def TFL_AddOp : TFL_Op<"add", [ TFL_RuntimePredOpTrait<"Operands should have valid shapes and element type needs to match", And<[CPred<"TFL::VerifyAddOpShapeConstraints(llvm::cast<AddOp>($_op))">, OperandsSameElementTypeConstraintBasePred]>>, ResultsBroadcastableShape, Pure, Commutative, QuantizableResult,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0)