Searched defs:tpu_function (Results 1 – 3 of 3) sorted by relevance
874 def tpu_function(sparse): function924 def tpu_function(sparse): function977 def tpu_function(sparse): function1053 def tpu_function(composite): function
1569 def tpu_function(args, kwargs): function
223 mlir::func::FuncOp tpu_function = xla_mesh->second.func(); in HandleCompilationOps() local