Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 374 for gotType (0.22 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

      StringAttr conv_padding = op.getPaddingAttr();
      SmallVector<int64_t> padding_nums;
      ShapedType lhs_shape = mlir::cast<ShapedType>(op.getLhs().getType());
      ShapedType rhs_shape = mlir::cast<ShapedType>(op.getRhs().getType());
    
      // Handle only static shape cases.
      // TODO(b/260284866): Handle dynamic shape cases.
      if (!lhs_shape.hasStaticShape()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/decompose_hybrid_quantization.cc

        SmallVector<Value> newOperands;
        newOperands.reserve(op->getNumOperands());
        for (auto operand : op->getOperands()) {
          if (QuantizedType::getQuantizedElementType(operand.getType())) {
            auto newTy = QuantizedType::castToExpressedType(operand.getType());
            newOperands.push_back(
                rewriter.create<TFL::DequantizeOp>(loc, newTy, operand));
            continue;
          }
    
          newOperands.push_back(operand);
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/utils/arithmetic_count_util.h

        auto weight = op->getOperand(1);
        auto weight_type =
            mlir::dyn_cast_or_null<mlir::RankedTensorType>(weight.getType());
        if (weight_type == nullptr || !weight_type.hasStaticShape()) return false;
    
        auto output = op->getResult(0);
        auto output_type =
            mlir::dyn_cast_or_null<mlir::RankedTensorType>(output.getType());
        if (output_type == nullptr || !output_type.hasStaticShape()) return false;
    
        int64_t cols = 1;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc

                 << "operand #" << i << " does not have a graph results to bind";
        if (graph.getResult(i).getType() != operand.getType()) {
          return fetch.emitOpError()
                 << "operand #" << i << " type mismatch graph results ("
                 << graph.getResult(i).getType() << " != " << operand.getType()
                 << ")";
        }
      }
      return success();
    }
    
    void GraphOp::print(OpAsmPrinter &p) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 42.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc

          auto current_type = quant::QuantizedType::getQuantizedElementType(
                                  quantize_output.getType())
                                  .getStorageType();
          if (current_type == input_type) {  // int8 == int8
            arg_type = quantize_output.getType();
            new_arg = block.addArgument(arg_type, loc);
            quantize_output.replaceAllUsesWith(new_arg);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.cc

      // etc.
      constexpr int64_t kLookupTableFindCostScale = 8;
      constexpr int64_t kLookupTableFindStringKeyCostScale = 16;
    
      auto value_type = mlir::cast<mlir::TensorType>(op.getValues().getType());
      auto key_type = mlir::cast<mlir::TensorType>(op.getKeys().getType());
    
      int64_t output_size = InferTensorSize(context, value_type);
    
      int64_t cost = kLookupTableFindCostScale * output_size;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td

      "TFL::IsBroadcastableElementsAttrAndType($0.getType(), $1.getType())">>;
    
    def OperandsBroadcastToOutputType : Constraint<CPred<
      "TFL::OperandsBroadcastToOutputType($0.getType(), $1.getType(), "
                                         "$2.getType())">>;
    
    def IsTailOfShape : Constraint<CPred<
      "TFL::IsTailOfShape($0.getType(), $1.getType())">>;
    
    def IsReducedTailOfShape : Constraint<CPred<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 20:31:41 UTC 2024
    - 66.4K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc

        return false;
      }
    
      auto opr1_type =
          llvm::dyn_cast_or_null<RankedTensorType>(op->getOperand(0).getType());
      auto opr2_type =
          llvm::dyn_cast_or_null<RankedTensorType>(op->getOperand(1).getType());
      auto res_type =
          llvm::dyn_cast_or_null<RankedTensorType>(op->getResult(0).getType());
    
      if (!(opr1_type && opr2_type && res_type)) {
        return false;
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc

        if (!input_dequant) return failure();
    
        if (!IsQI32Type(input_dequant.getType())) return failure();
    
        auto output_type =
            mlir::dyn_cast_or_null<ShapedType>(dequant_op.getOutput().getType());
        if (!output_type || !output_type.getElementType().isF32()) return failure();
    
        auto input_type = mlir::dyn_cast<ShapedType>(input_dequant.getType());
        // TODO(renjieliu): support UniformQuantizedPerAxisType.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc

          auto dcast = b.create<quantfork::DequantizeCastOp>(
              dq.getLoc(), dq.getOutput().getType(), dq.getInput());
          dq.getOutput().replaceAllUsesWith(dcast);
          dq.erase();
        } else if (auto q = llvm::dyn_cast<QuantizeOp>(op)) {
          auto qcast = b.create<quantfork::QuantizeCastOp>(
              q.getLoc(), q.getOutput().getType(), q.getInput());
          q.getOutput().replaceAllUsesWith(qcast);
          q.erase();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 02:50:01 UTC 2024
    - 3.5K bytes
    - Viewed (0)
Back to top