Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for resource_subtype (0.15 sec)

  1. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

          llvm::SmallVector<int64_t, 4> resource_subtype_shape(
              resource_shape.begin(), resource_shape.end());
          auto resource_subtype =
              GetTypeFromTFTensorShape(resource_subtype_shape, element_type);
          auto resource_type =
              mlir::TF::ResourceType::get({resource_subtype}, builder.getContext());
    
          auto tensor_type = mlir::cast<mlir::TensorType>(mlir_arg.getType());
          if (tensor_type.hasRank()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

      );
    
      DerivedTypeAttr dtype = DerivedTypeAttr<
          "return getElementTypeOrSelf(resource_subtype());">;
      DerivedAttr shape = DerivedAttr<
          "ShapedType",
          "return resource_subtype().cast<ShapedType>();",
          [{ mlir::TF::ShapeAttr::get($_ctxt, $_self) }]>;
    
      let extraClassDeclaration = [{
        TensorType resource_subtype() { return resource_type().getSubtypes()[0]; }
    
        ResourceType resource_type() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

          value = read_op.getValue();
        } else {
          llvm_unreachable("unexpected operator type");
        }
    
        TensorType resource_subtype = mlir::cast<TensorType>(value.getType());
        ResourceType resource_type =
            ResourceType::get({resource_subtype}, op.getContext());
        UnrankedTensorType new_resource_type =
            UnrankedTensorType::get(resource_type);
    
    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