Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dgraph_pruning.cc110 llvm::ArrayRef<std::string> ops_to_preserve) { in GraphPruningPass()
190 llvm::ArrayRef<std::string> ops_to_preserve) { in CreateTFExecutorGraphPruningPass()
H A 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
/aosp_15_r20/external/pytorch/torch/export/
H A Dexported_program.py177 def _override_composite_implicit_decomp(ops_to_preserve, decomp_table, safe=True): argument