Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for getOutputShape (0.1 sec)

  1. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

    mlir::LogicalResult TransposeConvOp::verify() {
      TransposeConvOp op = *this;
      ShapedType output_type = op.getOutput().getType().cast<ShapedType>();
      ShapedType output_shape_type =
          op.getOutputShape().getType().cast<ShapedType>();
      if (output_type.hasRank() && output_shape_type.hasStaticShape()) {
        if (output_type.getRank() != output_shape_type.getDimSize(0)) {
          return op.emitOpError(llvm::formatv(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

                   PatternRewriter& rewriter) const override {
        rewriter.replaceOpWithNewOp<TFL::ReshapeOp>(op, op.getOperand(),
                                                    op.getOutputShape());
      }
    };
    
    // Rewrites quantized stablehlo.select to tfl.select_v2.
    // TODO: b/322428814 - Add StableHLO quantizer integration tests for ODML.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
Back to top