- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for getEquation (0.23 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
} // unary einsum op is only supported to the case where the operation can be // replaced using reduce_sum and/or transpose if (const auto dnums_or = GetEinsumDimensionNumbersUnary(op.getEquation(), lhs)) { return rewriteToReduceSumAndTranspose(op, dnums_or.value(), rewriter); } return rewriter.notifyMatchFailure(op, "unsupported einsum lowering"); } #define GEN_PASS_DEF_TRANSFORMEINSUMPASS
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0)