Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for ComplexType (0.17 sec)

  1. tensorflow/compiler/mlir/tensorflow/utils/convert_type.cc

        case DT_BFLOAT16:
          *type = builder.getBF16Type();
          return absl::OkStatus();
        case DT_COMPLEX64:
          *type = mlir::ComplexType::get(builder.getF32Type());
          return absl::OkStatus();
        case DT_COMPLEX128:
          *type = mlir::ComplexType::get(builder.getF64Type());
          return absl::OkStatus();
        case tensorflow::DT_FLOAT8_E4M3FN:
          *type = builder.getFloat8E4M3FNType();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 7.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/convert_type.cc

          return builder.getI1Type();
        case tflite::TensorType_INT16:
          return builder.getIntegerType(16);
        case tflite::TensorType_COMPLEX64:
          return mlir::ComplexType::get(builder.getF32Type());
        case tflite::TensorType_COMPLEX128:
          return mlir::ComplexType::get(builder.getF64Type());
        case tflite::TensorType_INT4:
          return builder.getIntegerType(4);
        case tflite::TensorType_INT8:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/unfold_splat_constant_pass.cc

        if (!splat_elements_attr) {
          return;
        }
        if (splat_elements_attr.getNumElements() == 1) {
          return;
        }
        auto element_type = splat_elements_attr.getType().getElementType();
        if (mlir::isa<ComplexType>(element_type) ||
            mlir::isa<quant::QuantizedType>(element_type)) {
          return;
        }
        op_builder->setInsertionPoint(const_op);
        Value scalar = op_builder->create<mhlo::ConstantOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/rewrite_util.h

      } else if (auto int_ty = mlir::dyn_cast<IntegerType>(ty)) {
        IntegerAttr attr = IntegerAttr::get(int_ty, raw_value);
        return DenseElementsAttr::get(scalar_ty, attr);
      } else if (auto complex_ty = mlir::dyn_cast<ComplexType>(ty)) {
        Type complex_element_ty = complex_ty.getElementType();
        if (complex_element_ty.isF32()) {
          return DenseElementsAttr::get(
              scalar_ty, static_cast<std::complex<float>>(raw_value));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/utils/constant_utils.cc

        return DenseElementsAttr::get<float>(shaped_type,
                                             static_cast<float>(value));
      } else if (auto complex_type =
                     mlir::dyn_cast<mlir::ComplexType>(element_type)) {
        auto etype = complex_type.getElementType();
        if (etype.isF32()) {
          tensorflow::TensorProto repr;
          repr.set_dtype(tensorflow::DT_COMPLEX64);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.5K bytes
    - Viewed (0)
Back to top