- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for isSupportedAxis (0.26 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize.cc
if (!log_op || !log_op->hasOneUse()) { return failure(); } auto sum_op = dyn_cast_or_null<TFL::SumOp>(log_op.getX().getDefiningOp()); if (!sum_op || !sum_op.getKeepDims() || !isSupportedAxis( sum_op.getAxes(), mlir::cast<ShapedType>(sum_op.getOperand(0).getType()).getRank())) { return failure(); } if (!sum_op->hasOneUse()) { return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0)