- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TensorFlowTypeWithSubtype (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
rhs_element_type_with_subtype.GetSubtypes().size() == 1) { auto lhs_element_type_with_subtype = mlir::dyn_cast<TF::TensorFlowTypeWithSubtype>(lhs_element_type); TensorType subtype; if (!lhs_element_type_with_subtype) { DCOMMENT( "Unexpected inferred `TensorFlowTypeWithSubtype` when original " "result isn't"); } else if (lhs_element_type_with_subtype.GetSubtypes().size() > 1) {
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/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
supported_ops->insert(allowlist_ops.begin(), allowlist_ops.end()); } bool IsStringType(Type type) { if (mlir::isa<mlir::TF::StringType>(type)) return true; auto sub_type = mlir::dyn_cast<mlir::TF::TensorFlowTypeWithSubtype>(type); if (!sub_type) return false; bool has_string = llvm::any_of(sub_type.GetSubtypes(), [](TensorType type) { return mlir::isa<mlir::TF::StringType>(type.getElementType()); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0)