Searched refs:indices_ty (Results 1 – 2 of 2) sorted by relevance
110 auto indices_ty = op.indices().getType().dyn_cast<RankedTensorType>(); in Verify() local111 if (indices_ty && in Verify()112 !(axis == -1 || (axis >= 0 && axis <= indices_ty.getShape().size()))) { in Verify()115 << indices_ty.getShape().size() << "]"; in Verify()153 auto indices_ty = indices.getType().dyn_cast<RankedTensorType>(); in InferOneHotOpType() local154 if (!indices_ty) return unranked_ty; in InferOneHotOpType()156 auto shape = llvm::to_vector<2>(indices_ty.getShape()); in InferOneHotOpType()2230 auto indices_ty = op.indices().getType().dyn_cast<RankedTensorType>(); in Verify() local2231 if (!tensor_ty || !indices_ty) return success(); in Verify()2233 int64_t num_index_dims = indices_ty.getShape().back(); in Verify()
3924 auto indices_ty = op.indices().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local3927 if (!tensor_ty || !indices_ty || !updates_ty) return failure(); in matchAndRewrite()3931 int64_t num_index_dims = indices_ty.getShape().back(); in matchAndRewrite()3935 int64_t indices_rank = indices_ty.getRank(); in matchAndRewrite()4492 auto indices_ty = op.indices().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local4493 if (!indices_ty || !indices_ty.hasStaticShape()) return failure(); in matchAndRewrite()4494 ArrayRef<int64_t> indices_shape = indices_ty.getShape(); in matchAndRewrite()4495 Type element_type = indices_ty.getElementType(); in matchAndRewrite()