Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 22 for GetSubtypes (0.27 sec)

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

          .create<TF::ReadVariableOp>(
              loc,
              ArrayRef<Type>{mlir::cast<TF::ResourceType>(
                                 getElementTypeOrSelf(local_var.getType()))
                                 .getSubtypes()[0]},
              ArrayRef<Value>{local_var})
          .getValue();
    }
    
    // Creates an AssignVariableOp on a local variable.
    TF::AssignVariableOp WriteLocalVariable(Value local_var, Value value,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      auto variant_type =
          getElementTypeOrSelf(getOperand().getType()).cast<TF::VariantType>();
      if (variant_type.getSubtypes().empty()) return {};
      return ConvertShapeToAttr(variant_type.getSubtypes()[0], width);
    }
    
    //===----------------------------------------------------------------------===//
    // TensorListStackOp
    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/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

        Type type, llvm::SmallVector<int64_t, 8>* shape) {
      auto variant_type =
          mlir::dyn_cast<TF::VariantType>(getElementTypeOrSelf(type));
      if (!variant_type || variant_type.getSubtypes().size() != 1) return failure();
      TensorType tensor_type = variant_type.getSubtypes().front();
      if (!tensor_type.hasStaticShape()) return failure();
      for (auto d : tensor_type.getShape()) shape->push_back(d);
      return success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc

        return size.emitOpError("unknown tensor array");
      }
      auto buffer_type = getElementTypeOrSelf(local_var.getType())
                             .cast<TF::ResourceType>()
                             .getSubtypes()[0]
                             .cast<RankedTensorType>();
      OpBuilder builder(size);
      auto result = cutil::CreateScalarConst(buffer_type.getDimSize(0), builder,
                                             size.getLoc());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 02 20:41:19 UTC 2023
    - 40.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc

        // Get the data type of the resource.
        auto subtypes = mlir::cast<TF::ResourceType>(
                            getElementTypeOrSelf(execute.getOperand(it->second)))
                            .getSubtypes();
        if (subtypes.size() != 1) continue;
        auto data_type = getElementTypeOrSelf(subtypes[0]);
        // The XLA backend does not yet support formatting 64-bit data types.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc

      if (mlir::isa<mlir::TF::StringType>(type)) return true;
    
      auto sub_type = mlir::dyn_cast<mlir::TF::TensorFlowTypeWithSubtype>(type);
      if (!sub_type) return false;
    
      bool has_string = llvm::any_of(sub_type.GetSubtypes(), [](TensorType type) {
        return mlir::isa<mlir::TF::StringType>(type.getElementType());
      });
      return has_string;
    }
    
    bool HasStringOperand(Operation& op) {
      for (auto operand : op.getOperands()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

      if (auto resource_type =
              mlir::dyn_cast<mlir::TF::ResourceType>(arg_type.getElementType())) {
        llvm::ArrayRef<mlir::TensorType> subtypes = resource_type.getSubtypes();
        if (!subtypes.empty()) {
          AttrValue handle_dtypes_attr;
          AttrValue handle_shapes_attr;
          for (mlir::TensorType subtype : subtypes) {
            DataType dtype;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

      if (auto resource_type =
              mlir::dyn_cast<mlir::TF::ResourceType>(arg_type.getElementType())) {
        llvm::ArrayRef<mlir::TensorType> subtypes = resource_type.getSubtypes();
        if (!subtypes.empty()) {
          AttrValue handle_dtypes_attr;
          AttrValue handle_shapes_attr;
          for (mlir::TensorType subtype : subtypes) {
            DataType dtype;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

          "return getElementTypeOrSelf(element_type());">;
    
      let extraClassDeclaration = [{
        // Returns type of the TensorList element produced by this op.
        TensorType element_type() { return handle_dtype().getSubtypes()[0]; }
    
        // Returns data type of the result handle. Returned type contains type of
        // the TensorList element as a subtype.
        VariantType handle_dtype() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      }
    
      // We only support up to one nested type in tf_type.variant_type.
      if (variant_type.getSubtypes().size() > 1) {
        return std::nullopt;
      }
      if (variant_type.getSubtypes().empty()) {
        return variant_params;
      }
      mlir::TensorType tensor_type = variant_type.getSubtypes().front();
      tflite::TensorType tflite_element_type =
          GetTFLiteType(tensor_type.getElementType()).value();
    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