Home
last modified time | relevance | path

Searched defs:module_op (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtf_xla_mlir_translate.cc241 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHloViaBuilder()
294 mlir::ModuleOp module_op, llvm::raw_ostream& output, bool via_builder) { in MlirTfToHloTextTranslateFunctionImpl()
330 mlir::ModuleOp module_op, llvm::raw_ostream& output) { in MlirTfGraphToHloTextTranslateFunction()
389 mlir::ModuleOp module_op, llvm::raw_ostream& output) { in MlirModuleToSerializedMlirStringAttrTranslate()
398 mlir::ModuleOp module_op, llvm::raw_ostream& output) { in MlirTfToHloTextTranslateFunction()
403 mlir::ModuleOp module_op, llvm::raw_ostream& output) { in MlirTfToHloTextViaBuilderTranslateFunction()
Dcompile_mlir_util.cc341 Status LegalizeToHlo(mlir::ModuleOp module_op, llvm::StringRef device_type, in LegalizeToHlo()
375 Status BuildHloFromTfInner(mlir::ModuleOp module_op, xla::XlaBuilder& builder, in BuildHloFromTfInner()
389 mlir::ModuleOp module_op, llvm::StringRef device_type, in ConvertMLIRToXlaComputation()
407 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirSetup()
421 Status BuildHloFromTf(mlir::ModuleOp module_op, xla::XlaBuilder& builder, in BuildHloFromTf()
447 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in PopulateResultIOInfo()
468 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHlo()
519 mlir::ModuleOp module_op, llvm::ArrayRef<XlaArgument> args) { in RewriteWithArgs()
588 mlir::ModuleOp module_op, llvm::ArrayRef<XlaArgument> args, in CompileGraphSetup()
616 Status BuildHloFromModule(mlir::ModuleOp module_op, xla::XlaBuilder& builder, in BuildHloFromModule()
[all …]
Dserialize_mlir_module_utils.cc27 std::string SerializeMlirModule(mlir::ModuleOp module_op) { in SerializeMlirModule()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcluster_tf_ops_pass.cc178 void CreateFunctions(ModuleOp module_op, in CreateFunctions()
304 ModuleOp module_op = getOperation(); in runOnOperation() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dtfr_decompose_ctx.cc115 auto module_op = module.release(); in GetFromText() local
/external/llvm-project/mlir/lib/Bindings/Python/
DIRModules.cpp2761 PyOperationRef module_op = PyOperation::forOperation( in populateIRSubmodule() local