Searched refs:perm_op (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 140 auto perm_op = rewriter.create<ConstantOp>(loc, perm_type, perm_attr); in createTransposeOp() local
151 return rewriter.create<TF::TransposeOp>(loc, transposed_type, value, perm_op); in createTransposeOp()
|
| D | einsum.cc | 60 auto perm_op = rewriter->create<ConstantOp>(loc, perm_type, perm_attr); in createTransposeOp() local
68 perm_op); in createTransposeOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 94 auto perm_op = CreateI32DenseConst(builder, perm, location); in Transpose() local
106 value_to_transpose, perm_op); in Transpose()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 422 auto perm_op = rewriter.create<TF::ConstOp>(loc, perm_type, perm_attr); in legalizeFilter() local
433 return rewriter.create<TF::TransposeOp>(loc, result_type, filter, perm_op); in legalizeFilter()
|