Home
last modified time | relevance | path

Searched defs:mlir_context (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dbridge_logger_test.cc52 mlir::MLIRContext mlir_context(mlir_registry); in TEST() local
81 mlir::MLIRContext mlir_context(mlir_registry); in TEST() local
121 mlir::MLIRContext mlir_context(mlir_registry); in TEST() local
H A Dserialize_mlir_module_utils.cc37 mlir::MLIRContext* mlir_context, in DeserializeMlirModule()
H A Dcompile_mlir_util.cc674 mlir::MLIRContext mlir_context(mlir_registry); in CompileSerializedMlirToXlaHlo() local
862 mlir::MLIRContext& mlir_context, llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromGraph()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_context.h44 mlir::MLIRContext* mlir_context, llvm::Module* llvm_module) in IrEmitterContext()
70 mlir::MLIRContext* mlir_context() { return mlir_context_; } in mlir_context() function
H A Dgpu_compiler.cc1007 mlir::MLIRContext mlir_context(registry); in CompileModuleToLlvmIrImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Ddot_op_emitter.cc251 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in DotOpEmitter()
1241 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitNonBatchDotOperation()
1326 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in PotentiallyImplementedAsEigenMatmul()
1382 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitBatchDotOperation()
1508 llvm::IRBuilder<>* b, mlir::MLIRContext* mlir_context, in EmitDotOperation()
H A Dcpu_compiler.cc896 mlir::MLIRContext& mlir_context) { in LowerMLIRModule()
1000 HloModule* module, mlir::MLIRContext& mlir_context, in createMLIRModule()
1125 mlir::MLIRContext mlir_context; in CompileLegacyCpuExecutable() local
1406 mlir::MLIRContext mlir_context; in CompileAheadOfTime() local
H A Dir_emitter.cc95 IrEmitter::IrEmitter(mlir::MLIRContext* mlir_context, in IrEmitter()
/aosp_15_r20/external/tensorflow/tensorflow/tools/tfg_graph_transforms/
H A Dtfg_graph_transforms_main.cc132 mlir::MLIRContext* mlir_context) { in ImportModel()
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/const_dedupe_hoist/
H A Dpass.cc75 MLIRContext* mlir_context; member