Searched refs:unranked_ty (Results 1 – 2 of 2) sorted by relevance
167 auto unranked_ty = UnrankedTensorType::get(element_ty); in InferOneHotOpType() local168 if (axis_val < -1) return unranked_ty; in InferOneHotOpType()171 if (!indices_ty) return unranked_ty; in InferOneHotOpType()979 auto unranked_ty = UnrankedTensorType::get(element_ty); in InferSelectV2OpType() local983 if (!broadcasted_ty) return unranked_ty; in InferSelectV2OpType()987 if (!cond_ranked_ty || !broadcasted_ranked_ty) return unranked_ty; in InferSelectV2OpType()995 return unranked_ty; in InferSelectV2OpType()
2497 auto unranked_ty = UnrankedTensorType::get(element_ty); in InferExpandDimsOpType() local2500 if (!input_ty) return unranked_ty; in InferExpandDimsOpType()2505 return unranked_ty; in InferExpandDimsOpType()2509 if (dim_val < -input_rank - 1 || dim_val > input_rank + 1) return unranked_ty; in InferExpandDimsOpType()