Home
last modified time | relevance | path

Searched refs:eraseArgument (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_op_lifting_cleanup.cc178 for (FuncOp func : cloned_branches) func.eraseArgument(idx); in EliminateUnusedResultsForIfCase()
228 cloned_cond.eraseArgument(idx); in EliminateUnusedResultsForWhile()
230 cloned_body.eraseArgument(idx); in EliminateUnusedResultsForWhile()
391 body.eraseArgument(result_idx); in CanonicalizeWhileRegion()
392 cond.eraseArgument(result_idx); in CanonicalizeWhileRegion()
Dpromote_resources_to_args.cc333 block.eraseArgument(argument_index_to_remove); in PromoteResourcesToArguments()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dpost_quantize.cc81 bb.eraseArgument(0); in RemoveQuantizationAdaptorOps()
98 bb.eraseArgument(0); in RemoveQuantizationAdaptorOps()
/external/llvm-project/mlir/lib/IR/
DBlock.cpp174 void Block::eraseArgument(unsigned index) { in eraseArgument() function in Block
193 eraseArgument(originalNumArgs - i - 1); in eraseArguments()
/external/llvm-project/mlir/include/mlir/IR/
DRegion.h108 void eraseArgument(unsigned index) { front().eraseArgument(index); } in eraseArgument() function
DBlock.h104 void eraseArgument(unsigned index);
DFunctionSupport.h264 void eraseArgument(unsigned argIndex) { eraseArguments({argIndex}); } in eraseArgument() function
/external/llvm-project/mlir/lib/Transforms/
DNormalizeMemRefs.cpp379 funcOp.front().eraseArgument(argIndex); in normalizeFuncOpMemRefs()
385 funcOp.front().eraseArgument(argIndex + 1); in normalizeFuncOpMemRefs()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc183 block->eraseArgument(i); in ReplaceBlockArgs()
Dlegalize_tf_communication.cc519 block.eraseArgument(0); in UpdateControlFlowBlockArgWithToken()
/external/llvm-project/mlir/lib/Transforms/Utils/
DRegionUtils.cpp333 block.eraseArgument(e - i - 1); in deleteDeadness()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc582 for (int idx : llvm::reverse(args_to_erase)) main_fn.eraseArgument(idx); in RewriteWithArgs()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2737 body_block.eraseArgument(arg_index); in matchAndRewrite()
2738 cond_block.eraseArgument(arg_index); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc3061 new_cond_block.eraseArgument(next_idx); in matchAndRewrite()
3062 new_body_block.eraseArgument(next_idx); in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/Linalg/IR/
DLinalgOps.cpp1813 payload.eraseArgument(bbArgBaseOffset + reversed); in matchAndRewrite()