Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 18 of 18 for GetSubtypes (0.22 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc

    // no single type in the resource.
    Type GetResourceSubtype(Value value) {
      auto resource_type =
          mlir::dyn_cast<TF::ResourceType>(getElementTypeOrSelf(value.getType()));
      auto subtypes = resource_type.getSubtypes();
      if (subtypes.size() == 1) return subtypes[0];
      return nullptr;
    }
    
    // Replaces all `tf.VarIsInitializedOp` in a block with a constant true.
    // TODO(b/171039585): Replace this with proper analysis of
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      // This is required to populate derived attributes during export in a
      // meaningful way. Else during export to GraphDef element_type() query
      // will result in out of bounds access/assert.
      if (handle_dtype().getSubtypes().size() != 1) {
        return emitOpError(
            "must have exactly one subtype in the result variant type");
      }
    
      if (!IsOfRankOrUnranked(op.getElementShape(), 0) &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
Back to top