| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | mlir.cc | 58 StatusOr<XlaComputation> PyMlirModuleToXlaComputation(std::string mlir_module, in PyMlirModuleToXlaComputation()
87 py::module mlir_module = m.def_submodule("mlir", "MLIR/XLA integration"); in BuildMlirSubmodule() local
|
| H A D | py_client.cc | 337 std::string mlir_module, CompileOptions options, in CompileMlir()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/ |
| H A D | jit_compiler.cc | 141 std::string_view mlir_module, in JitCompiler()
158 std::string_view mlir_module, in Instantiate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/runtime/ |
| H A D | jit_executable.cc | 97 std::string_view mlir_module, std::string_view entrypoint, Options opts, in Instantiate()
149 JitExecutable::JitExecutable(std::string_view mlir_module, in JitExecutable()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | serialize_mlir_module_utils.cc | 38 mlir::OwningOpRef<mlir::ModuleOp>* mlir_module) { in DeserializeMlirModule()
|
| H A D | compile_mlir_util.cc | 675 mlir::OwningOpRef<mlir::ModuleOp> mlir_module; in CompileSerializedMlirToXlaHlo() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | set_tpu_infeed_layout.cc | 31 bool SetTPUInfeedLayout(mlir::OwningOpRef<mlir::ModuleOp> &mlir_module) { in SetTPUInfeedLayout()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compile_op_impl.h | 38 TpuCompileOpKernelImpl(const std::string& mlir_module, in TpuCompileOpKernelImpl()
|
| H A D | tpu_compile.proto | 51 string mlir_module = 8; field
|
| H A D | tpu_compile_op_impl.cc | 67 std::string mlir_module; in CreateMlirImpl() local
|
| H A D | tpu_compile_op_support.h | 50 absl::string_view mlir_module; member
|
| H A D | tpu_compile_op_common.h | 66 TpuCompileOpKernelCommon(const std::string& mlir_module, in TpuCompileOpKernelCommon()
|
| H A D | tpu_compile_op_support.cc | 396 std::string* mlir_module) { in CompileOpMetadataFromContext()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/jit/ |
| H A D | tf_jitrt_benchmark_test.cc | 29 static const char* mlir_module = R"( variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| H A D | conv_emitter_test.cc | 57 mlir::OwningOpRef<mlir::ModuleOp> mlir_module( in CompileHloConvAndGetMlir() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_compiler.h | 55 const std::string& mlir_module, in JitRtAotCompilationResult()
|
| H A D | gpu_compiler.cc | 879 mlir::ModuleOp mlir_module, llvm::StringRef entry_function_name, in LowerToJitRt()
1008 mlir::OwningOpRef<mlir::ModuleOp> mlir_module = in CompileModuleToLlvmIrImpl() local
|
| H A D | gpu_executable.cc | 1027 absl::string_view mlir_module, in LoadFromObjFile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_remaining_ops.cc | 124 mlir::OwningOpRef<mlir::ModuleOp>* mlir_module) { in GetHostFunc()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | mlir_emitter.cc | 117 mlir::OwningOpRef<mlir::ModuleOp> mlir_module = mlir::ModuleOp::create(loc); in EmitMlirFuncAndCall() local
|
| H A D | cpu_compiler.cc | 895 Status LowerMLIRModule(mlir::ModuleOp mlir_module, in LowerMLIRModule()
1004 auto mlir_module = builder.create<mlir::ModuleOp>(builder.getUnknownLoc()); in createMLIRModule() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/ |
| H A D | tf_cpurt_executor.cc | 70 const std::string& mlir_module, const std::string& entrypoint, in Compile()
|
| H A D | tf_jitrt_executor.cc | 78 const std::string& mlir_module, const std::string& entrypoint, in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| H A D | tpu_client_extension.cc | 155 -> StatusOr<std::unique_ptr<PyTpuExecutable>> { in PYBIND11_MODULE()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo.proto | 771 string mlir_module = 4; field
|