Home
last modified time | relevance | path

Searched refs:ModuleOp (Results 1 – 25 of 426) sorted by relevance

12345678910>>...18

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dpasses.h31 std::unique_ptr<OperationPass<ModuleOp>> CreateBreakUpIslandsPass();
66 std::unique_ptr<OperationPass<ModuleOp>>
71 std::unique_ptr<OperationPass<ModuleOp>>
80 std::unique_ptr<OperationPass<ModuleOp>> CreateTFShapeInferencePass();
89 std::unique_ptr<OperationPass<ModuleOp>> CreateGuaranteeAllFuncsOneUsePass();
127 std::unique_ptr<OperationPass<ModuleOp>> createConvertTfControlFlowToScfPass();
163 std::unique_ptr<OperationPass<ModuleOp>> CreateResourceDeviceInferencePass();
171 std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteResourcesToArgsPass(
176 std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteVarHandlesToArgsPass();
199 std::unique_ptr<OperationPass<ModuleOp>> CreateStackOpsDecompositionPass();
[all …]
H A Dtest_passes.h28 std::unique_ptr<OperationPass<ModuleOp>> CreateFreezeVariableTestPass();
42 std::unique_ptr<OperationPass<ModuleOp>> CreateTestSideEffectAnalysisPass();
44 std::unique_ptr<OperationPass<ModuleOp>> CreateTestResourceAliasAnalysisPass();
46 std::unique_ptr<OperationPass<ModuleOp>> CreateInitTextFileToImportTestPass();
47 std::unique_ptr<OperationPass<ModuleOp>>
51 std::unique_ptr<OperationPass<ModuleOp>> CreateLiftVariablesTestPass();
52 std::unique_ptr<OperationPass<ModuleOp>>
56 std::unique_ptr<OperationPass<ModuleOp>>
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dcreate_dtensor_mlir_passes.h66 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
69 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
72 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
75 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
78 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
81 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
84 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
87 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
90 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
93 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
[all …]
H A DPasses.td118 : Pass<"dtensor-handle_cross_cluster_dependences", "mlir::ModuleOp"> {
126 : Pass<"dtensor-annotate-global-shape", "mlir::ModuleOp"> {
134 : Pass<"dtensor-layout-propagation-v2", "mlir::ModuleOp"> {
142 : Pass<"dtensor-mesh-propagation", "mlir::ModuleOp"> {
150 : Pass<"dtensor-spmd-expansion", "mlir::ModuleOp"> {
158 : Pass<"dtensor-all-reduce-lowering", "mlir::ModuleOp"> {
166 : Pass<"dtensor-reduce-scatter-lowering", "mlir::ModuleOp"> {
174 : Pass<"dtensor-all-gather-lowering", "mlir::ModuleOp"> {
182 : Pass<"dtensor-all-scatter-lowering", "mlir::ModuleOp"> {
190 : Pass<"dtensor-cluster-function-conversion", "mlir::ModuleOp"> {
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dpasses.h46 class ModuleOp; variable
76 std::unique_ptr<OperationPass<ModuleOp>> CreateLowerStaticTensorListPass(
80 std::unique_ptr<OperationPass<ModuleOp>> CreateLowerStaticTensorListPass();
129 std::unique_ptr<OperationPass<ModuleOp>> CreateTrimFunctionsPass();
131 std::unique_ptr<OperationPass<ModuleOp>> CreateTrimFunctionsPass(
136 std::unique_ptr<OperationPass<ModuleOp>> CreatePrepareCompositeFunctionsPass();
142 std::unique_ptr<OperationPass<ModuleOp>> CreateOptimizeFunctionalOpsPass();
168 std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTFWhilePass();
174 std::unique_ptr<OperationPass<ModuleOp>> CreateWhileOutlinePass();
192 std::unique_ptr<OperationPass<ModuleOp>>
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Ddevice_util_test.cc67 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
68 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
106 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
107 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
115 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
116 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
124 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
125 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
135 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
136 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
[all …]
H A Ddump_mlir_util_test.cc32 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
33 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
42 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
43 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
52 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
53 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
76 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
77 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
88 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
89 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
[all …]
H A Dtpu_rewrite_device_util_test.cc631 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
632 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
649 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
650 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
668 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
669 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
687 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
688 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
704 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST()
705 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
[all …]
H A Dcompile_mlir_util.cc116 mlir::ModuleOp module, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in GetXlaInputShapes()
186 mlir::ModuleOp module, bool use_resource_updates_for_aliases, in GetOutputInfo()
327 mlir::ModuleOp module) { in RefineShapes()
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()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dimport_model.h43 stream_executor::port::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
52 stream_executor::port::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
60 stream_executor::port::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
67 stream_executor::port::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
76 stream_executor::port::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
92 stream_executor::port::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
145 stream_executor::port::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
153 std::string MlirModuleToString(mlir::ModuleOp module,
155 std::string MlirModuleToString(mlir::ModuleOp m, bool show_debug_info = false);
H A Dtf_mlir_translate.h42 StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> GraphdefToMlirTranslateFunction(
60 StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> GraphdefToMlirTranslateFunction(
72 StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
91 StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
105 StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> SavedModelObjectGraphToMlirImport(
115 StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> SavedModelSignatureDefsToMlirImport(
127 StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dpasses.h43 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
49 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
54 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
59 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>> CreateMergeTfIfOpsPass();
63 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
80 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
85 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
98 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
103 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
236 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/
H A Dops_test.cc31 OpT findOp(ModuleOp module) { in findOp()
58 OwningOpRef<ModuleOp> module = in TEST()
59 mlir::parseSourceString<mlir::ModuleOp>(code, &context); in TEST()
99 OwningOpRef<ModuleOp> module = in TEST()
100 mlir::parseSourceString<mlir::ModuleOp>(code, &context); in TEST()
144 OwningOpRef<ModuleOp> module = in TEST()
145 mlir::parseSourceString<mlir::ModuleOp>(code, &context); in TEST()
183 OwningOpRef<ModuleOp> module = in TEST()
184 mlir::parseSourceString<mlir::ModuleOp>(code, &context); in TEST()
H A Dtf_op_registry_test.cc54 OwningOpRef<ModuleOp> module = in TEST()
55 mlir::parseSourceString<mlir::ModuleOp>(code, &context); in TEST()
74 OwningOpRef<ModuleOp> module = in TEST()
75 mlir::parseSourceString<mlir::ModuleOp>(code, &context); in TEST()
100 OwningOpRef<ModuleOp> module = in TEST()
101 mlir::parseSourceString<mlir::ModuleOp>(code, &context); in TEST()
H A Dutility_test.cc40 OwningOpRef<mlir::ModuleOp> module = in TEST()
41 mlir::parseSourceString<mlir::ModuleOp>(code, &context); in TEST()
75 OwningOpRef<mlir::ModuleOp> module = in TEST()
76 mlir::parseSourceString<mlir::ModuleOp>(code, &context); in TEST()
110 OwningOpRef<mlir::ModuleOp> module = in TEST()
111 mlir::parseSourceString<mlir::ModuleOp>(code, &context); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Transforms/
H A Dpasses.h25 class ModuleOp; variable
82 std::unique_ptr<OperationPass<ModuleOp>> createComputeOpAndFuncBufferizePass();
86 std::unique_ptr<OperationPass<ModuleOp>> createFinalBufferizePass();
88 std::unique_ptr<OperationPass<ModuleOp>> createFinalBufferizePass(
95 std::unique_ptr<OperationPass<ModuleOp>>
107 std::unique_ptr<OperationPass<ModuleOp>> createOneShotBufferizePass();
109 std::unique_ptr<OperationPass<ModuleOp>> createGenericHostToLLVMPass();
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dlmhlo_to_jitrt.cc67 using mlir::ModuleOp;
207 SymbolTable sym_table(op->template getParentOfType<ModuleOp>()); in matchAndRewrite()
271 SymbolTable sym_table(op->getParentOfType<ModuleOp>()); in matchAndRewrite()
309 SymbolTable sym_table(op->getParentOfType<ModuleOp>()); in matchAndRewrite()
335 ModuleOp module = op->getParentOfType<ModuleOp>(); in matchAndRewrite()
415 ModuleOp module = op->template getParentOfType<ModuleOp>(); in matchAndRewrite()
475 ModuleOp module = op->template getParentOfType<ModuleOp>(); in matchAndRewrite()
573 ModuleOp module = op->template getParentOfType<ModuleOp>(); in matchAndRewrite()
869 SymbolTable sym_table(op->getParentOfType<ModuleOp>()); in matchAndRewrite()
906 GlobalConstantsArgs GetConstantArgs(ModuleOp m) { in GetConstantArgs()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
H A Dkernel_creator.cc110 Status LowerTFToJITInvocation(mlir::ModuleOp module, in LowerTFToJITInvocation()
140 Status LowerTFtoLoops(mlir::ModuleOp module, llvm::ArrayRef<int64_t> tile_sizes, in LowerTFtoLoops()
235 Status LowerLoopsToGPU(mlir::ModuleOp module, bool embed_memref_prints, in LowerLoopsToGPU()
308 Status LowerKernelBodiesToLowLevelIr(mlir::ModuleOp module, in LowerKernelBodiesToLowLevelIr()
353 Status AmendKernelLLVMIRWithStaticKnowledge(mlir::ModuleOp module, in AmendKernelLLVMIRWithStaticKnowledge()
369 Status GenerateDeviceCode(mlir::ModuleOp module, in GenerateDeviceCode()
390 Status LowerHostSideToFinalForm(mlir::ModuleOp module, bool apply_cl_options) { in LowerHostSideToFinalForm()
407 StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> SetupContextAndParseModule( in SetupContextAndParseModule()
416 mlir::OwningOpRef<mlir::ModuleOp> module = in SetupContextAndParseModule()
417 mlir::parseSourceString<mlir::ModuleOp>(tf_code, &context); in SetupContextAndParseModule()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dpasses.h32 std::unique_ptr<OperationPass<ModuleOp>> CreateInsertMainFunctionPass();
38 std::unique_ptr<OperationPass<ModuleOp>>
46 std::unique_ptr<OperationPass<ModuleOp>>
55 std::unique_ptr<OperationPass<ModuleOp>>
59 std::unique_ptr<OperationPass<ModuleOp>> CreateInsertQuantizedFunctionsPass(
69 std::unique_ptr<OperationPass<ModuleOp>> CreateQuantizeCompositeFunctionsPass(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dpasses.h36 std::unique_ptr<OperationPass<ModuleOp>> CreateEmbedTFFrameworkPass();
41 std::unique_ptr<OperationPass<ModuleOp>> CreateRewriteTFFrameworkAssert();
59 std::unique_ptr<OperationPass<ModuleOp>> CreateTFKernelToLLVMPass(
64 std::unique_ptr<OperationPass<ModuleOp>> CreateShapeToDescriptorsPass();
84 std::unique_ptr<OperationPass<ModuleOp>> CreateEmbedMemRefPrintsPass();
98 std::unique_ptr<OperationPass<ModuleOp>> CreateKernelgenFinalBufferizePass();
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dmlir_gpu_test_base.cc46 mlir::ModuleOp module, se::Stream* stream) { in CompileMlirModule()
75 mlir::ModuleOp module, se::Stream* stream, in RunMlirModule()
106 mlir::ModuleOp module, std::vector<absl::Span<uint8_t>> arguments) { in RunMlirModuleWithHostBuffers()
139 StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> MlirGpuTestBase::ParseMlirModule( in ParseMlirModule()
149 mlir::OwningOpRef<mlir::ModuleOp> module = in ParseMlirModule()
150 mlir::parseSourceString<mlir::ModuleOp>( in ParseMlirModule()
162 TF_ASSIGN_OR_RETURN(mlir::OwningOpRef<mlir::ModuleOp> module, in RunMlirTextWithHostBuffers()
170 TF_ASSIGN_OR_RETURN(mlir::OwningOpRef<mlir::ModuleOp> module, in CompileMlirText()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dxla_mlir_translate.cc54 mlir::OwningOpRef<mlir::ModuleOp> HloToMlirHloTranslateFunction( in HloToMlirHloTranslateFunction()
64 mlir::OwningOpRef<mlir::ModuleOp> module = in HloToMlirHloTranslateFunction()
65 mlir::ModuleOp::create(mlir::UnknownLoc::get(context)); in HloToMlirHloTranslateFunction()
76 mlir::OwningOpRef<mlir::ModuleOp> HloTextToMlirHloTranslateFunction( in HloTextToMlirHloTranslateFunction()
88 mlir::OwningOpRef<mlir::ModuleOp> module = in HloTextToMlirHloTranslateFunction()
89 mlir::ModuleOp::create(mlir::UnknownLoc::get(context)); in HloTextToMlirHloTranslateFunction()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/
H A Djax_to_tfl_flatbuffer.cc76 mlir::OwningOpRef<mlir::ModuleOp> HloToMlirHloTranslateFunction( in HloToMlirHloTranslateFunction()
86 mlir::OwningOpRef<mlir::ModuleOp> module = in HloToMlirHloTranslateFunction()
87 mlir::ModuleOp::create(mlir::UnknownLoc::get(context)); in HloToMlirHloTranslateFunction()
98 mlir::OwningOpRef<mlir::ModuleOp> HloTextToMlirHloTranslateFunction( in HloTextToMlirHloTranslateFunction()
111 mlir::OwningOpRef<mlir::ModuleOp> module = in HloTextToMlirHloTranslateFunction()
112 mlir::ModuleOp::create(mlir::UnknownLoc::get(context)); in HloTextToMlirHloTranslateFunction()
164 mlir::OwningOpRef<mlir::ModuleOp> module; in ConvertJaxToTFLiteFlatBuffer()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/transforms/
H A Dpasses.h26 class ModuleOp; variable
63 std::unique_ptr<OperationPass<ModuleOp>> createLegalizeToLhloPass();
67 std::unique_ptr<OperationPass<ModuleOp>> createLegalizeToMemrefPass();
70 std::unique_ptr<OperationPass<ModuleOp>> createLegalizeToArithmeticPass();
140 std::unique_ptr<OperationPass<ModuleOp>> createExpandHloTuplesPass(
148 std::unique_ptr<OperationPass<ModuleOp>> createConvertToSignlessPass();
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Dpasses.h41 std::unique_ptr<OperationPass<ModuleOp>> CreateRaiseTargetSubgraphsPass();
44 std::unique_ptr<OperationPass<ModuleOp>> CreateAlternativeSubgraphPass(
48 std::unique_ptr<OperationPass<ModuleOp>> CreateComputeCostPass();
51 std::unique_ptr<OperationPass<ModuleOp>> CreatePickSubgraphsPass();
60 std::unique_ptr<OperationPass<ModuleOp>> CreateFoldConstantsToSubgraphPass(

12345678910>>...18