- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for GetSubtypes (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.cc
if (resource_type.getSubtypes().size() == 1) return resource_type.getSubtypes().front(); return UnrankedTensorType::get(element_type); } static bool HasResourceSubtype(Value resource) { return resource.getType() .cast<TensorType>() .getElementType() .cast<ResourceType>() .getSubtypes() .size() == 1; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_variable_ops_to_ml_program.cc
if (!tensor) return nullptr; TF::ResourceType resource = llvm::dyn_cast<TF::ResourceType>(tensor.getElementType()); if (!resource || resource.getSubtypes().size() != 1) return nullptr; return resource.getSubtypes().front(); } else if (auto global_tensor_op = dyn_cast<tf_saved_model::GlobalTensorOp>(source)) { return global_tensor_op.getType(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
} auto element_type = getElementTypeOrSelf(first_operand_type); if (mlir::isa<TF::ResourceType>(element_type)) { first_operand_type = mlir::cast<TF::ResourceType>(element_type).getSubtypes().front(); } auto tensor_type = mlir::dyn_cast_or_null<TensorType>(first_operand_type); if (!(tensor_type && tensor_type.hasRank())) { return op->emitError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
auto arg_type = mlir::cast<RankedTensorType>(arg.getType()); assert(arg_type.getRank() == 0); llvm::ArrayRef<TensorType> underlying_type = mlir::cast<TF::ResourceType>(arg_type.getElementType()).getSubtypes(); // If the arg type already matches the global_tensor type, we don't need // to do anything. if (!underlying_type.empty() && underlying_type[0] == global_tensor.getType()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0)