Searched refs:perm_type (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 136 auto perm_type = RankedTensorType::get({static_cast<int32_t>(perm.size())}, in createTransposeOp() local
139 auto perm_attr = DenseElementsAttr::get(perm_type, llvm::makeArrayRef(perm)); in createTransposeOp()
140 auto perm_op = rewriter.create<ConstantOp>(loc, perm_type, perm_attr); in createTransposeOp()
|
| D | einsum.cc | 57 auto perm_type = RankedTensorType::get( in createTransposeOp() local
59 auto perm_attr = DenseElementsAttr::get(perm_type, permutation); in createTransposeOp()
60 auto perm_op = rewriter->create<ConstantOp>(loc, perm_type, perm_attr); in createTransposeOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2403 auto perm_type = op.perm().getType().dyn_cast<RankedTensorType>(); in Verify() local
2407 if (perm_type && perm_type.getRank() != 1) { in Verify()
2410 << perm_type.getRank(); in Verify()
2419 if (!x_type || !y_type || !perm_type || !perm_type.hasStaticShape()) { in Verify()
2423 if (x_type.getRank() != perm_type.getNumElements()) { in Verify()
2426 << perm_type.getNumElements() << ", and x of rank " in Verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 418 auto perm_type = RankedTensorType::get({static_cast<int>(perm.size())}, in legalizeFilter() local
421 DenseElementsAttr::get(perm_type, llvm::makeArrayRef<int>(perm)); in legalizeFilter()
422 auto perm_op = rewriter.create<TF::ConstOp>(loc, perm_type, perm_attr); in legalizeFilter()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2609 auto perm_type = op.perm().getType().cast<ShapedType>(); in Verify() local
2611 if (input_type.hasStaticShape() && perm_type.hasStaticShape()) { in Verify()
2612 if (perm_type.getNumElements() != input_type.getRank()) { in Verify()
|