- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ConvertTFEinsumOp (0.56 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc
// For the uniform opset, it is requested to maintain the BatchMatmul logic. // For the TF opset, since we need to test the effect we remain it as a // future work. patterns.add<TF::ConvertTFEinsumOp>(ctx); } if (failed(applyPatternsAndFoldGreedily(func, std::move(patterns)))) { func.emitError() << "quant-prepare-lifting failed."; signalPassFailure(); } } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
void TransformEinsumPass::runOnOperation() { RewritePatternSet patterns(&getContext()); auto func = getOperation(); patterns.add<ConvertTFEinsumOp>(&getContext()); (void)applyPatternsAndFoldGreedily(func, std::move(patterns)); } } // namespace LogicalResult ConvertTFEinsumOp::matchAndRewrite( TF::EinsumOp op, PatternRewriter& rewriter) const { if (op->getNumOperands() == 1) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
ReorderFakeQuantPattern<TF::TransposeOp>>(ctx); if (unfold_batch_matmul_) { TF::PopulateUnrollTfBatchMatMul(ctx, phase_2_patterns); } phase_2_patterns.add<TF::ConvertTFEinsumOp, ConvertTFStridedSlice, ConvertRfftToRfft2d, RemoveIdentity>(ctx); phase_2_patterns.add<ConvertTFConv2D, ConvertTFDepthwiseConv2dNative>( ctx, allow_bf16_and_f16_type_legalization_);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0)