Searched refs:unranked_ty (Results 1 – 2 of 2) sorted by relevance
150 auto unranked_ty = UnrankedTensorType::get(element_ty); in InferOneHotOpType() local151 if (axis_val < -1) return unranked_ty; in InferOneHotOpType()154 if (!indices_ty) return unranked_ty; in InferOneHotOpType()812 auto unranked_ty = UnrankedTensorType::get(element_ty); in InferSelectV2OpType() local816 if (!broadcasted_ty) return unranked_ty; in InferSelectV2OpType()820 if (!cond_ranked_ty || !broadcasted_ranked_ty) return unranked_ty; in InferSelectV2OpType()828 return unranked_ty; in InferSelectV2OpType()
2128 auto unranked_ty = UnrankedTensorType::get(element_ty); in InferExpandDimsOpType() local2131 if (!input_ty) return unranked_ty; in InferExpandDimsOpType()2136 return unranked_ty; in InferExpandDimsOpType()2140 if (dim_val < -input_rank - 1 || dim_val > input_rank + 1) return unranked_ty; in InferExpandDimsOpType()