Home
last modified time | relevance | path

Searched refs:FoldOperandsPermutation (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_op_interfaces.td97 "LogicalResult", "FoldOperandsPermutation",
Dtf_ops_a_m.cc2365 return ::mlir::TF::FoldOperandsPermutation(permutation, op); in InferenceFoldOperandsPermutation()
2388 LogicalResult FusedBatchNormV2Op::FoldOperandsPermutation( in FoldOperandsPermutation() function in mlir::TF::FusedBatchNormV2Op
2401 LogicalResult FusedBatchNormV3Op::FoldOperandsPermutation( in FoldOperandsPermutation() function in mlir::TF::FusedBatchNormV3Op
2721 LogicalResult MaxPoolOp::FoldOperandsPermutation( in FoldOperandsPermutation() function in mlir::TF::MaxPoolOp
2723 return ::mlir::TF::FoldOperandsPermutation( in FoldOperandsPermutation()
2775 LogicalResult MeanOp::FoldOperandsPermutation(ArrayRef<int64_t> permutation) { in FoldOperandsPermutation() function in mlir::TF::MeanOp
Dtf_ops_helpers.inc386 // See `FoldOperandsTransposeInterface::FoldOperandsPermutation`.
388 LogicalResult FoldOperandsPermutation(
Dtf_ops_n_z.cc309 LogicalResult PadOp::FoldOperandsPermutation(ArrayRef<int64_t> permutation) { in FoldOperandsPermutation() function in mlir::TF::PadOp
Dtf_generated_ops.td4744 LogicalResult FoldOperandsPermutation(ArrayRef<int64_t> permutation);
4796 LogicalResult FoldOperandsPermutation(ArrayRef<int64_t> permutation);
8078 LogicalResult FoldOperandsPermutation(ArrayRef<int64_t> permutation);
8287 LogicalResult FoldOperandsPermutation(ArrayRef<int64_t> permutation);
9362 LogicalResult FoldOperandsPermutation(ArrayRef<int64_t> permutation);
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlayout_optimization.cc369 if (failed(fold_operands.FoldOperandsPermutation(permutation))) return; in MoveTransposeAfter()