Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for resultType (0.21 sec)

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

                               llvm::function_ref<APFloat(APFloat)> calculate) {
      assert(IsF32ShapedType(result_type) || IsBF16ShapedType(result_type));
      auto result_shape_type = result_type.cast<ShapedType>();
    
      if (!result_shape_type.hasStaticShape()) return {};
    
      if (auto dense_elements = operand.dyn_cast_or_null<DenseElementsAttr>()) {
        SmallVector<APFloat, 16> new_values;
    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/ir/tfl_ops.td

    def TFL_BroadcastableBinaryBuilder :
      OpBuilder<(ins "Value":$lhs, "Value":$rhs),
      [{
        auto resultType =
          OpTrait::util::getBroadcastedType(lhs.getType(), rhs.getType());
        if (!resultType)
          mlir::emitError($_state.location, "non-broadcastable operands");
        $_state.addOperands({lhs, rhs});
        $_state.types.push_back(resultType);
      }]>;
    
    def TFL_FusedBroadcastableBinaryBuilder :
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

        Type operand_type = std::get<0>(entry).getType();
        Value result = std::get<1>(entry);
        TensorType result_type = mlir::cast<TensorType>(result.getType());
        Type inferred_type = TypeMeet(result_type, operand_type);
        if (result_type == inferred_type) continue;
    
        if (!UpdateTypeAndInsertIncompatibleUseCasts(inferred_type, result))
          continue;
        changed = true;
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

      auto matmul = rewriter.create<TF::BatchMatMulV3Op>(
          loc, RankedTensorType::get(matmul_shape, result_type.getElementType()),
          lhs_flattend, rhs_flattend);
    
      if (result_type.hasStaticShape()) {
        auto reshaped =
            rewriter.create<mhlo::ReshapeOp>(loc, result_type, matmul.getResult());
        return reshaped.getResult();
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

                           Value y, BoolAttr incompatible_shape_error) {
      auto result_type = DeduceEqualCmpOpType(&builder, result.location, x, y,
                                              incompatible_shape_error);
      return build(builder, result, result_type, x, y, incompatible_shape_error);
    }
    
    //===----------------------------------------------------------------------===//
    // OneHotOp
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      for (auto it : llvm::zip(op.getResultTypes(), updated_results)) {
        auto& updated_result = std::get<1>(it);
        Type result_type = std::get<0>(it);
        if (result_type != updated_result.getType()) {
          updated_result =
              rewriter.create<TF::CastOp>(op.getLoc(), result_type, updated_result,
                                          /*Truncate=*/rewriter.getBoolAttr(false));
        }
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/optimize.cc

        auto indices_value = gather_nd_first.getIndices();
    
        auto result_type = dyn_cast<RankedTensorType>(result_value.getType());
        auto params_type = dyn_cast<RankedTensorType>(params_value.getType());
        auto indices_type = dyn_cast<RankedTensorType>(indices_value.getType());
    
        // Handle static shape cases only.
        if (!result_type || !params_type || !indices_type) return failure();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

        return success();
      }
    
      void rewrite(stablehlo::UniformDequantizeOp op,
                   PatternRewriter& rewriter) const override {
        rewriter.replaceOpWithNewOp<TFL::DequantizeOp>(
            op, /*resultTypes=*/op->getResultTypes(), /*input=*/op.getOperand());
      }
    };
    
    // Rewrites `stablehlo.dot_general` to `tfl.fully_connected` or
    // `tfl.batch_matmul` when it accepts uniform quantized tensors.
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

          os << operand_type;
        }
      }
      os << ") -> (";
      if (!inst->getResultTypes().empty()) {
        bool first = true;
        for (Type result_type : inst->getResultTypes()) {
          os << (!first ? ", " : "");
          first = false;
          os << result_type;
        }
      }
      os << ")";
      // Print out attributes except for large elementsattributes (which should
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
Back to top