Searched refs:GetSubtypes (Results 1 – 5 of 5) sorted by relevance
191 ArrayRef<TensorType> TensorFlowTypeWithSubtype::GetSubtypes() { in GetSubtypes() function in mlir::TF::TensorFlowTypeWithSubtype230 auto lhs_wst_st = lhs_wst.GetSubtypes(); in BroadcastCompatible()231 auto rhs_wst_st = rhs_wst.GetSubtypes(); in BroadcastCompatible()309 auto a_wst_st = a_wst.GetSubtypes(); in GetCastCompatibleType()310 auto b_wst_st = b_wst.GetSubtypes(); in GetCastCompatibleType()
245 ArrayRef<TensorType> GetSubtypes();
123 GetSubtypes(Type type) { in GetSubtypes() function252 handle_shapes_and_types[index] = GetSubtypes(operand_type); in InferReturnTypeComponentsForTFOp()
86 return tws.GetSubtypes().empty() || in CanBeRefined()87 llvm::any_of(tws.GetSubtypes(), CanBeRefined); in CanBeRefined()109 !element_type_with_subtype.GetSubtypes().empty(); in CanRefineTypeWith()972 if (operand_handle_type.GetSubtypes().empty() || in GetElementTypeFromOperand()973 !result_handle_type.GetSubtypes().empty()) in GetElementTypeFromOperand()
155 bool has_string = llvm::any_of(sub_type.GetSubtypes(), [](TensorType type) { in IsStringType()