Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetDimensionSizeOp (0.24 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc

    using ::mlir::stablehlo::ConvolutionOp;
    using ::mlir::stablehlo::DotGeneralOp;
    using ::mlir::stablehlo::DynamicBroadcastInDimOp;
    using ::mlir::stablehlo::GatherOp;
    using ::mlir::stablehlo::GetDimensionSizeOp;
    using ::mlir::stablehlo::ReshapeOp;
    using ::mlir::stablehlo::UniformQuantizeOp;
    using ::stablehlo::quantization::Method;
    using ::stablehlo::quantization::QuantizedDimension;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 06:04:36 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

        rewriter.eraseOp(op);
        return success();
      }
    };
    
    // Converts a MHLO::GetDimensionSizeOp to TF ops.
    class ConvertGetDimensionSizeOp
        : public OpConversionPattern<mhlo::GetDimensionSizeOp> {
     public:
      using OpConversionPattern::OpConversionPattern;
    
      LogicalResult matchAndRewrite(
          mhlo::GetDimensionSizeOp op, OpAdaptor adaptor,
          ConversionPatternRewriter& rewriter) const final {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
Back to top