Searched refs:indices_ety (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 3210 auto indices_ety = indices_ty.getElementType(); in matchAndRewrite() local
3245 indices_ety = rewriter.getI32Type(); in matchAndRewrite()
3246 indices_ty = RankedTensorType::get(indices_ty.getShape(), indices_ety); in matchAndRewrite()
3250 RankedTensorType::get({result_rank}, indices_ety); in matchAndRewrite()
3260 rewriter, loc, UnrankedTensorType::get(indices_ety), multiply_op, in matchAndRewrite()
3269 rewriter, loc, UnrankedTensorType::get(indices_ety), reduce_op, in matchAndRewrite()
|