- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for ConvertTensorProto (0.26 sec)
-
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) -
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) -
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) -
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) -
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) -
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)