| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | compile_mlir_util.cc | 460 Status LegalizeToHlo(mlir::ModuleOp module_op, llvm::StringRef device_type, in LegalizeToHlo()
501 Status BuildHloFromTfInner(mlir::ModuleOp module_op, xla::XlaBuilder& builder, in BuildHloFromTfInner()
517 mlir::ModuleOp module_op, llvm::StringRef device_type, in ConvertMLIRToXlaComputation()
553 Status CompileMlirSetup(mlir::ModuleOp module_op, in CompileMlirSetup()
564 Status BuildHloFromTf(mlir::ModuleOp module_op, xla::XlaBuilder& builder, in BuildHloFromTf()
587 Status PopulateCollectiveInfo(mlir::ModuleOp module_op, in PopulateCollectiveInfo()
615 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in PopulateResultIOInfo()
636 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHlo()
695 mlir::ModuleOp module_op, llvm::ArrayRef<XlaArgument> args) { in RewriteWithArgs()
765 mlir::ModuleOp module_op, llvm::ArrayRef<XlaArgument> args, in CompileGraphSetup()
[all …]
|
| H A D | tf_xla_mlir_translate.cc | 255 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHloViaBuilder()
310 mlir::ModuleOp module_op, llvm::raw_ostream& output, bool via_builder) { in MlirTfToHloTextTranslateFunctionImpl()
345 mlir::ModuleOp module_op, llvm::raw_ostream& output) { in MlirTfGraphToHloTextTranslateFunction()
408 mlir::ModuleOp module_op, llvm::raw_ostream& output) { in MlirModuleToSerializedMlirStringAttrTranslate()
417 mlir::ModuleOp module_op, llvm::raw_ostream& output) { in MlirTfToHloTextTranslateFunction()
422 mlir::ModuleOp module_op, llvm::raw_ostream& output) { in MlirTfToHloTextViaBuilderTranslateFunction()
|
| H A D | serialize_mlir_module_utils.cc | 27 std::string SerializeMlirModule(mlir::ModuleOp module_op) { in SerializeMlirModule()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| H A D | kernel_ops_pattern.cc | 147 mlir::OwningOpRef<mlir::ModuleOp> module_op = in CloneToModule() local
274 auto module_op = CloneToModule(op, arguments, get_global_ops); in Match() local
459 mlir::ModuleOp module_op, mlir::PatternRewriter& rewriter) const { in matchAndRewrite()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | cluster_tf_ops_pass.cc | 179 void CreateFunctions(ModuleOp module_op, in CreateFunctions()
305 ModuleOp module_op = getOperation(); in runOnOperation() local
|
| H A D | canonicalize_compile_and_replicate_attributes.cc | 46 ModuleOp module_op = func_op->getParentOfType<ModuleOp>(); in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/ |
| H A D | execution_metadata_exporter_test.cc | 115 auto module_op = module.get(); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/tfg_graph_transforms/ |
| H A D | tfg_graph_transforms_main.cc | 154 tensorflow::Status ExportTFGModule(mlir::ModuleOp module_op, in ExportTFGModule()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/ |
| H A D | tf_op_wrapper_test.cc | 46 Operation *module_op = module.get(); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| H A D | tfr_decompose_ctx.cc | 117 auto module_op = module.release(); in GetFromText() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/ |
| H A D | quantize_model.cc | 98 const mlir::ModuleOp module_op) { in ConvertMlirModuleToGraphDef()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| H A D | raise_target_subgraphs.cc | 196 Subgraph* subgraph, OpBuilder* builder, ModuleOp module_op, in BuildFuncOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | save_restore_spmd_expander.cc | 848 mlir::ModuleOp module_op = op->getParentOfType<mlir::ModuleOp>(); in ComputeLayoutForward() local
|