- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for VariantType (0.13 sec)
-
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
mlir::Type type, const std::string& name); // Builds TF::VariantType from the given element type. Returns std::nullopt if // failure. Returns empty vector if the element type is not TF::VariantType or // there is empty TensorType in the TF::VariantType. std::optional<std::vector<BufferOffset<tflite::VariantSubType>>> BuildTFVariantType(mlir::Type element_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
TF_RETURN_IF_ERROR( ConvertDataType(element_dtype->type(), builder, &etype)); } return GetTypeFromTFTensorShape( {}, mlir::TF::VariantType::get({mlir::UnrankedTensorType::get(etype)}, etype.getContext())); } if (node.IsWhileNode()) { auto* output_shapes = node.attrs().Find("output_shapes");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
DCOMMENT("InferShapeForListInitOps " << *op << " could be inferred " << element_type); if (!element_type || !element_type.hasStaticShape()) return false; auto variant_type = VariantType::get(element_type, op->getContext()); auto tensor_type = tensorflow::GetTypeFromTFTensorShape({}, variant_type); bool changed = RefineResultType(op, handle, tensor_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
OpFoldResult TensorListElementShapeOp::fold(FoldAdaptor) { int width = getType().cast<ShapedType>().getElementType().getIntOrFloatBitWidth(); auto variant_type = getElementTypeOrSelf(getOperand().getType()).cast<TF::VariantType>(); if (variant_type.getSubtypes().empty()) return {}; return ConvertShapeToAttr(variant_type.getSubtypes()[0], width); } //===----------------------------------------------------------------------===//
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)