Searched refs:size_ty (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_framework_legalize_to_llvm.cc | 61 Location loc, Type size_ty, Type element_ty, in ConvertArrayAttrToStackAllocatedArray() argument
70 loc, size_ty, rewriter->getIntegerAttr(size_ty, 0)); in ConvertArrayAttrToStackAllocatedArray()
78 loc, size_ty, rewriter->getIntegerAttr(size_ty, array_attr.size())); in ConvertArrayAttrToStackAllocatedArray()
83 loc, size_ty, rewriter->getIntegerAttr(size_ty, e.index())); in ConvertArrayAttrToStackAllocatedArray()
93 Location loc, Type size_ty, Type element_ty, in ConvertIntegerArrayAttrToStackAllocatedArray() argument
96 assert(size_ty.isa<IntegerType>() && "expect integer size type"); in ConvertIntegerArrayAttrToStackAllocatedArray()
99 loc, size_ty, element_ty, attr, rewriter, [&](Attribute attr) { in ConvertIntegerArrayAttrToStackAllocatedArray()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1247 RankedTensorType size_ty = GetRankedTensorTypeForOperand(op.size()); in verify() local
1248 if (size_ty && size_ty.getRank() != 1) { in verify()
1252 if (!begin_ty || !size_ty || !begin_ty.hasStaticShape() || in verify()
1253 !size_ty.hasStaticShape()) in verify()
1256 if (begin_ty.getNumElements() != size_ty.getNumElements()) { in verify()
|