Search Options

Results per page
Sort
Preferred Languages
Advance

Results 51 - 60 of 123 for input_dtype (0.26 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc

        if (!reshape_type.hasStaticShape()) return failure();
        ArrayRef<int64_t> reshape_shape = reshape_type.getShape();
    
        auto input_type = mlir::cast<ShapedType>(op.getInput().getType());
        auto output_type = mlir::cast<ShapedType>(op.getOutput().getType());
    
        if (!input_type.hasRank() || !output_type.hasRank()) return failure();
    
        // The pattern attempts to reduce the rank of the input to BroadcastTo.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

          return failure();
    
        Value input = tf_op.getInput();
        RankedTensorType input_type =
            mlir::dyn_cast<RankedTensorType>(input.getType());
        // Only rank size four input will be only available by the tf.Conv2D
        // operator verification.
        if (!input_type || input_type.isDynamicDim(3)) {
          return failure();
        }
        // Check if the given op is based on grouped convolution.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  3. subprojects/core/src/main/java/org/gradle/api/internal/initialization/transform/MergeInstrumentationAnalysisTransform.java

            File input = getInput().get().getAsFile();
            InstrumentationInputType inputType = getInputType(input);
            switch (inputType) {
                case DEPENDENCY_ANALYSIS_DATA:
                    doMergeAndOutputAnalysis(input, outputs);
                    return;
                case ORIGINAL_ARTIFACT:
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Thu Apr 18 15:08:33 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/common/ir/QuantOps.cc

      for (auto input : llvm::zip(getOperandTypes(), getInputSpecs())) {
        Type inputType = std::get<0>(input);
        Attribute inputSpec = std::get<1>(input);
        if (!isValidQuantizationSpec(inputSpec, inputType)) {
          return emitOpError() << "has incompatible specification " << inputSpec
                               << " and input type " << inputType;
        }
      }
    
      for (auto result : llvm::zip(getResultTypes(), getOutputSpecs())) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/quantization/ir/QuantOps.cc

      for (auto input : llvm::zip(getOperandTypes(), getInputSpecs())) {
        Type inputType = std::get<0>(input);
        Attribute inputSpec = std::get<1>(input);
        if (!isValidQuantizationSpec(inputSpec, inputType)) {
          return emitOpError() << "has incompatible specification " << inputSpec
                               << " and input type " << inputType;
        }
      }
    
      for (auto result : llvm::zip(getResultTypes(), getOutputSpecs())) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h

        if (dyn_cast_or_null<quantfork::QuantizeCastOp>(next_op)) return failure();
    
        auto input_type = mlir::cast<ShapedType>(transpose_op.getInput().getType());
        auto perm_type = mlir::cast<ShapedType>(transpose_op.getPerm().getType());
        if (input_type.hasStaticShape() && perm_type.hasStaticShape()) {
          if (perm_type.getNumElements() != input_type.getRank()) {
            return transpose_op.emitOpError(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 28K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

          return failure();
        }
    
        const auto input_type = op.getLhs().getType().cast<TensorType>();
        if (!(input_type.getRank() == 2 || input_type.getRank() == 3)) {
          LLVM_DEBUG(llvm::dbgs() << "Input expected to have rank of 2 or 3. Got: "
                                  << input_type << ".\n");
          return failure();
        }
    
        const Value filter = op.getRhs();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

               << ", and input of rank " << input_type.getRank();
      }
    
      if (input_type && output_type) {
        if (input_type.getRank() != output_type.getRank()) {
          return op.emitOpError()
                 << "expected rank of input to equal to rank of output"
                 << ", got input of rank " << input_type.getRank()
                 << ", and output of rank " << output_type.getRank();
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc

      BoolAttr narrow_range = builder.getBoolAttr(false);
    
      auto add_quantize_op = [&](Location loc, Type input_type, Block* block,
                                 Block::iterator insertion_point, Value arg,
                                 int i) {
        if (auto shaped = mlir::dyn_cast<ShapedType>(input_type)) {
          if (mlir::isa<FloatType>(shaped.getElementType())) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

            break;
          default:
            return nullptr;  // Not yet supported
        }
      } else {
        return nullptr;  // Not yet supported
      }
    
      input_type = input_type.clone(new_storage_type);
      return input_type;
    }
    
    // Replaces quant.qcast op to composite quantize_i8 function.
    class ReplaceQuantizePattern
        : public mlir::OpRewritePattern<quantfork::QuantizeCastOp> {
     public:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
Back to top