- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for resource_subtype (0.15 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
llvm::SmallVector<int64_t, 4> resource_subtype_shape( resource_shape.begin(), resource_shape.end()); auto resource_subtype = GetTypeFromTFTensorShape(resource_subtype_shape, element_type); auto resource_type = mlir::TF::ResourceType::get({resource_subtype}, builder.getContext()); auto tensor_type = mlir::cast<mlir::TensorType>(mlir_arg.getType()); if (tensor_type.hasRank()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
); DerivedTypeAttr dtype = DerivedTypeAttr< "return getElementTypeOrSelf(resource_subtype());">; DerivedAttr shape = DerivedAttr< "ShapedType", "return resource_subtype().cast<ShapedType>();", [{ mlir::TF::ShapeAttr::get($_ctxt, $_self) }]>; let extraClassDeclaration = [{ TensorType resource_subtype() { return resource_type().getSubtypes()[0]; } ResourceType resource_type() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
value = read_op.getValue(); } else { llvm_unreachable("unexpected operator type"); } TensorType resource_subtype = mlir::cast<TensorType>(value.getType()); ResourceType resource_type = ResourceType::get({resource_subtype}, op.getContext()); UnrankedTensorType new_resource_type = UnrankedTensorType::get(resource_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)