Home
last modified time | relevance | path

Searched refs:user_module (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtpu_test_wrapper.py192 user_module = import_user_module() variable
198 move_test_classes_into_scope(user_module)
199 run_user_main(user_module)
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dtfr_decompose_ctx.cc181 Status TFRDecomposeContext::DecomposeGraph(mlir::ModuleOp user_module) { in DecomposeGraph() argument
183 if (failed(pm_.run(user_module))) { in DecomposeGraph()
216 Status DecomposeGraph(mlir::ModuleOp user_module) { in DecomposeGraph() argument
217 mlir::MLIRContext* mlir_ctx = user_module.getContext(); in DecomposeGraph()
219 return ctx->DecomposeGraph(user_module); in DecomposeGraph()
Dtfr_decompose_ctx.h57 Status DecomposeGraph(mlir::ModuleOp user_module);
76 Status DecomposeGraph(mlir::ModuleOp user_module);