- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for ConvertScalarTypeToDataType (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/convert_type.cc
} Status ConvertToDataType(Type type, DataType* dtype) { if (auto stype = mlir::dyn_cast<ShapedType>(type)) { TF_RETURN_IF_ERROR( ConvertScalarTypeToDataType(stype.getElementType(), dtype)); } else { TF_RETURN_IF_ERROR(ConvertScalarTypeToDataType(type, dtype)); } return absl::OkStatus(); } void ConvertToMlirShape(const TensorShape& input_shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_type.h
// Converts the TensorFlow DataType 'dtype' into an MLIR (scalar) type. Status ConvertDataType(DataType dtype, mlir::Builder builder, mlir::Type* type); // Converts a scalar MLIR type to a TensorFlow Datatype. Status ConvertScalarTypeToDataType(mlir::Type type, DataType* dtype); // Converts an MLIR type to TensorFlow DataType. If 'type' is a scalar type, it // is converted directly. If it is a shaped type, the element type is converted.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/saved_model/saved_model.cc
} return ""; } absl::StatusOr<std::pair<tensorflow::DataType, tensorflow::PartialTensorShape>> ProcessTensorSpec(mlir::TensorType type) { tensorflow::DataType dtype; TF_RETURN_IF_ERROR( ConvertScalarTypeToDataType(type.getElementType(), &dtype)); if (!type.hasRank()) return std::make_pair(dtype, tensorflow::PartialTensorShape()); auto shape = type.getShape(); llvm::SmallVector<int64_t, 4> dims;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
AttrValueMap* values) { AttrValue value; auto& type_list = *value.mutable_list(); for (auto type : types) { DataType dtype; TF_RETURN_IF_ERROR(ConvertScalarTypeToDataType(type, &dtype)); type_list.add_type(dtype); } auto result = values->insert({string(name), value}); assert(result.second && "cannot have multiple attributes with the same name"); (void)result;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0)