Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for ConvertTensorProto (0.38 sec)

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

          return mlir::TypeAttr::get(type);
        }
        case AttrValue::kShape:
          return ConvertTensorShapeProto(value.shape(), builder->getContext());
        case AttrValue::kTensor:
          return ConvertTensorProto(value.tensor(), builder);
        case AttrValue::kList: {
          absl::InlinedVector<mlir::Attribute, 8> attrs;
          for (const auto& item : value.list().i())
            attrs.push_back(builder->getI64IntegerAttr(item));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 4.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h

    #include "tensorflow/core/protobuf/struct.pb.h"
    
    namespace tensorflow {
    
    using tsl::StatusOr;
    
    // Converts an TensorFlow tensor proto into an MLIR elements attribute.
    absl::StatusOr<mlir::ElementsAttr> ConvertTensorProto(
        const TensorProto& input_tensor, mlir::Builder* builder);
    
    // Converts an TensorFlow tensor into an MLIR elements attribute.
    absl::StatusOr<mlir::ElementsAttr> ConvertTensor(const Tensor& input_tensor,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 2.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/utils/convert_tensor_test.cc

      tensor.add_float_val(42.0);
    
      mlir::MLIRContext context;
      mlir::Builder builder(&context);
      TF_ASSERT_OK_AND_ASSIGN(mlir::ElementsAttr attribute,
                              ConvertTensorProto(tensor, &builder));
      EXPECT_THAT(
          attribute,
          AllOf(Eq(mlir::DenseElementsAttr::get(
                    mlir::RankedTensorType::get({1ULL << 35}, builder.getF32Type()),
                    42.0f)),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc

        // when DenseElementsAttr is updated.
        case DT_COMPLEX64:
        case DT_COMPLEX128:
        default:
          return -1;
      }
    }
    
    absl::StatusOr<ElementsAttr> ConvertTensorProto(const TensorProto& input_tensor,
                                                    Builder* builder) {
      // If there is only one actual element in the proto, but its shape would
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 20.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

            buffer.size() / (shaped_type.getElementTypeBitWidth() / CHAR_BIT));
      }
      TF_ASSIGN_OR_RETURN(mlir::ElementsAttr compressed_data,
                          tensorflow::ConvertTensorProto(repr, &builder));
    
      const int dim_metadata_size = tensor.sparsity->dim_metadata.size();
      std::vector<mlir::TFL::DimensionMetadataAttr> dim_metadata(dim_metadata_size);
      for (int i = 0; i < dim_metadata_size; i++) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

          shape_inference::InferenceContext* context, mlir::Builder builder);
    
      // Converts the tensor proto into an MLIR elements attribute.
      absl::StatusOr<mlir::ElementsAttr> ConvertTensorProto(
          const TensorProto& value) {
        return ::tensorflow::ConvertTensorProto(value, &builder_);
      }
    
      // Converts func name in graphdef to mlir::SymbolRefAttribute.
      absl::StatusOr<mlir::FlatSymbolRefAttr> ConvertFunctionCallName(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top