Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for ConvertTFBatchMatMulOp (0.37 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc 

    void mlir::TF::PopulateUnrollTfBatchMatMul(MLIRContext* context,
                                           RewritePatternSet& patterns) {
  patterns.add<ConvertTFBatchMatMulOp<TF::BatchMatMulOp>,
               ConvertTFBatchMatMulOp<TF::BatchMatMulV2Op>,
               ConvertTFBatchMatMulOp<TF::BatchMatMulV3Op>>(context);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.6K bytes
    - Viewed (0)
  3. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc 

          /*keep_num_dims=*/rewriter.getBoolAttr(true),
      /*asymmetric_quantize_inputs=*/mlir::BoolAttr());
  rewriter.replaceOp(op, {fc_op.getResult(0)});

  return true;
}

LogicalResult ConvertTFBatchMatMulOp::matchAndRewrite(
    Operation* op, PatternRewriter& rewriter) const {
  if (ConvertTFBatchMatMulOp2TFLFullyConnectedOp<TF::BatchMatMulOp>(op,
                                                                    rewriter))
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 20:06:54 UTC 2024
    - 45.2K bytes
    - Viewed (0)
Back to top