- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetSubtypes (0.23 sec)
-
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
return type; } if (!variant_ty.getSubtypes().empty()) { // Short-circut if the variant type has subtype info. return UnrankedTensorType::get( variant_ty.getSubtypes()[0].getElementType()); } Type value_type = value.getType(); Type element_type; variant_ty = value_type.dyn_cast<TF::VariantType>(); if (variant_ty && !variant_ty.getSubtypes().empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
"return getElementTypeOrSelf(element_type());">; let extraClassDeclaration = [{ // Returns type of the TensorList element produced by this op. TensorType element_type() { return handle_dtype().getSubtypes()[0]; } // Returns data type of the result handle. Returned type contains type of // the TensorList element as a subtype. VariantType handle_dtype() {
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/resource_op_lifting.cc
// no single type in the resource. Type GetResourceSubtype(Value value) { auto resource_type = mlir::dyn_cast<TF::ResourceType>(getElementTypeOrSelf(value.getType())); auto subtypes = resource_type.getSubtypes(); if (subtypes.size() == 1) return subtypes[0]; return nullptr; } // Replaces all `tf.VarIsInitializedOp` in a block with a constant true. // TODO(b/171039585): Replace this with proper analysis of
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0)