Home
last modified time | relevance | path

Searched defs:ops_to_preserve (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgraph_pruning.cc110 llvm::ArrayRef<std::string> ops_to_preserve) { in GraphPruningPass()
190 llvm::ArrayRef<std::string> ops_to_preserve) { in CreateTFExecutorGraphPruningPass()
Dbridge.cc84 const llvm::SmallVector<std::string, 4> ops_to_preserve = { in CreateTPUBridgePipelineImpl() local
304 const llvm::SmallVector<std::string, 4> ops_to_preserve = {}; in CreateTFXLABridgePipeline() local
/external/pytorch/torch/export/
Dexported_program.py177 def _override_composite_implicit_decomp(ops_to_preserve, decomp_table, safe=True): argument