Searched defs:orig_func (Results 1 – 4 of 4) sorted by relevance
154 def __new__(cls, orig_func=None): argument
912 def autocast_func(orig_func): argument919 def autocast_func_cuda(orig_func): argument926 def autocast_func_cpu(orig_func): argument
833 uptr orig_func; in OverrideFunction() local
3347 auto orig_func = symbol_table.lookup<mlir::func::FuncOp>( in CreateSavedModelIR() local