Searched refs:perm_attr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | layout_optimization.cc | 160 auto perm_attr = [&](Permutation permutation) -> DenseIntElementsAttr { in runOnFunction() local
171 auto arg_perm = builder.create<ConstOp>(loc, perm_attr(args_permutation)); in runOnFunction()
180 auto res_perm = builder.create<ConstOp>(loc, perm_attr(res_permutation)); in runOnFunction()
233 auto perm_attr = permutation_op.value().cast<DenseElementsAttr>(); in MoveTransposeBefore() local
234 for (const auto& value : perm_attr.getIntValues()) in MoveTransposeBefore()
|
| D | unroll_batch_matmul.cc | 139 auto perm_attr = DenseElementsAttr::get(perm_type, llvm::makeArrayRef(perm)); in createTransposeOp() local
140 auto perm_op = rewriter.create<ConstantOp>(loc, perm_type, perm_attr); in createTransposeOp()
|
| D | einsum.cc | 59 auto perm_attr = DenseElementsAttr::get(perm_type, permutation); in createTransposeOp() local
60 auto perm_op = rewriter->create<ConstantOp>(loc, perm_type, perm_attr); in createTransposeOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 420 auto perm_attr = in legalizeFilter() local
422 auto perm_op = rewriter.create<TF::ConstOp>(loc, perm_type, perm_attr); in legalizeFilter()
|