- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for ConvertTypeToTensorType (0.18 sec)
-
tensorflow/compiler/mlir/lite/utils/convert_type.h
#include "tensorflow/core/framework/types.pb.h" namespace mlir { class Builder; } // namespace mlir namespace tflite { // Convert the MLIR type to the corresponding TFLite tensor. tflite::TensorType ConvertTypeToTensorType(mlir::Type type); // Convert the scalar type of a TFlite tensor to the corresponding MLIR type. mlir::Type ConvertElementType(tflite::TensorType type, mlir::Builder builder);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc
tflite::TensorType tflite_type = tflite::ConvertTypeToTensorType(reserve.getElementDtype()); return CreateListReserveOptions(context, tflite_type); } if (auto empty = llvm::dyn_cast_or_null<mlir::TF::EmptyTensorListOp>(op)) { tflite::TensorType tflite_type = tflite::ConvertTypeToTensorType(empty.getElementDtype()); return CreateListReserveOptions(context, tflite_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/convert_type.cc
#include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/platform/errors.h" namespace tflite { using absl::StatusOr; namespace errors = tensorflow::errors; tflite::TensorType ConvertTypeToTensorType(mlir::Type type) { if (type.isF16()) { return tflite::TensorType_FLOAT16; } else if (type.isBF16()) { return tflite::TensorType_BFLOAT16; } else if (type.isF32()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
llvm_unreachable("invalid mirror_pad_enum in conversion."); } static tflite::TensorType ConvertDerivedTypeAttrForOptionWriter( mlir::Type type, flatbuffers::FlatBufferBuilder* builder) { return tflite::ConvertTypeToTensorType(type); } // I32Attr already returns an int as required by flatbuffer builders. static int ConvertI32AttrForOptionWriter( int i, flatbuffers::FlatBufferBuilder* builder) { return i; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0)