Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for Dtype (0.05 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

            VLOG(1) << "[potentially conservative] Op type `" << node.type_string()
                    << "` is stateful but effects not modelled";
          } else {
            // See if any resource type is used.
            bool resource = false;
            std::function<bool(mlir::Type)> record_resource;
            record_resource = [&](mlir::Type type) {
              type.walk([&](mlir::Type t) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

              dtype == tensorflow::DT_UINT16 || dtype == tensorflow::DT_INT16 ||
              dtype == tensorflow::DT_UINT8 || dtype == tensorflow::DT_INT8 ||
              dtype == tensorflow::DT_HALF || dtype == tensorflow::DT_BFLOAT16 ||
              dtype == tensorflow::DT_FLOAT || dtype == tensorflow::DT_DOUBLE ||
              dtype == tensorflow::DT_COMPLEX64 ||
              dtype == tensorflow::DT_COMPLEX128 || dtype == tensorflow::DT_BOOL) {
            return {};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  3. tensorflow/c/c_api.cc

                metadata.type = TF_ATTR_INT;
              } else if (typestr == "list(float)") {
                metadata.type = TF_ATTR_FLOAT;
              } else if (typestr == "list(bool)") {
                metadata.type = TF_ATTR_BOOL;
              } else if (typestr == "list(type)") {
                metadata.type = TF_ATTR_TYPE;
              } else if (typestr == "list(shape)") {
                metadata.type = TF_ATTR_SHAPE;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

                        Attribute value) {
      // Handle the case where the type and value are already tensors.
      if (mlir::isa<TensorType>(type) && mlir::isa<ElementsAttr>(value)) {
        result.addTypes(type);
        result.addAttribute("value", value);
        return;
      }
    
      // Otherwise, default to the attribute builder.
      ConstOp::build(builder, result, value);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

        }
        if (ftype.isF64()) {
          return tflite::TensorType_COMPLEX128;
        }
        return Status(absl::StatusCode::kInvalidArgument, "Unsupported type");
      } else if (auto itype = mlir::dyn_cast<mlir::IntegerType>(type)) {
        switch (itype.getWidth()) {
          case 1:
            return tflite::TensorType_BOOL;
          case 4:
            if (itype.isUnsigned()) {
              return Status(absl::StatusCode::kInvalidArgument,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

      else
        return UnrankedTensorType::get(type.getValue());
    }
    }  // namespace
    
    // Returns whether type can be further refined.
    bool CanBeRefined(Type type) {
      auto shape_type = mlir::dyn_cast<ShapedType>(type);
      if (!shape_type) return false;
    
      // Returns whether type with subtypes can be further refined.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
Back to top