Home
last modified time | relevance | path

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

12345678910>>...14

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dpasses.h29 std::unique_ptr<OperationPass<ModuleOp>> CreateBreakUpIslandsPass();
52 std::unique_ptr<OperationPass<ModuleOp>>
57 std::unique_ptr<OperationPass<ModuleOp>>
65 std::unique_ptr<OperationPass<ModuleOp>> CreateTFShapeInferencePass();
68 std::unique_ptr<OperationPass<ModuleOp>> CreateGuaranteeAllFuncsOneUsePass();
129 std::unique_ptr<OperationPass<ModuleOp>> CreateResourceDeviceInferencePass();
136 std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteResourcesToArgsPass();
140 std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteVarHandlesToArgsPass();
158 std::unique_ptr<OperationPass<ModuleOp>> CreateStackOpsDecompositionPass();
162 std::unique_ptr<OperationPass<ModuleOp>> CreateTensorListOpsDecompositionPass();
[all …]
Dtf_saved_model_passes.h28 std::unique_ptr<OperationPass<ModuleOp>> CreateOptimizeGlobalTensorsPass();
31 std::unique_ptr<OperationPass<ModuleOp>> CreateFreezeGlobalTensorsPass(
41 std::unique_ptr<OperationPass<ModuleOp>>
47 std::unique_ptr<OperationPass<ModuleOp>> CreateLiftVariablesPass(
Doptimize_global_tensors.cc43 : public PassWrapper<OptimizeGlobalTensorsPass, OperationPass<ModuleOp>> {
69 explicit ResourceAnalyzer(ModuleOp module) { in ResourceAnalyzer()
202 GlobalTensorUsesMap CreateGlobalTensorUsesMap(ModuleOp module) { in CreateGlobalTensorUsesMap()
228 ModuleOp module, const GlobalTensorUsesMap& global_tensor_uses_map, in MarkGlobalTensorsImmutable()
239 void EraseUnusedGlobalTensors(ModuleOp module, in EraseUnusedGlobalTensors()
256 void EraseUnusedBoundInputs(ModuleOp module) { in EraseUnusedBoundInputs()
293 std::unique_ptr<OperationPass<ModuleOp>> CreateOptimizeGlobalTensorsPass() { in CreateOptimizeGlobalTensorsPass()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dpasses.h25 class ModuleOp; variable
48 std::unique_ptr<OperationPass<ModuleOp>> CreateLowerStaticTensorListPass();
64 std::unique_ptr<OperationPass<ModuleOp>> CreateTrimFunctionsPass(
69 std::unique_ptr<OperationPass<ModuleOp>> CreatePrepareCompositeFunctionsPass();
75 std::unique_ptr<OperationPass<ModuleOp>> CreateOptimizeFunctionalOpsPass();
87 std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTFWhilePass();
90 std::unique_ptr<OperationPass<ModuleOp>> CreateWhileOutlinePass();
100 std::unique_ptr<OperationPass<ModuleOp>>
105 std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeVariablesPass();
109 std::unique_ptr<OperationPass<ModuleOp>> CreateRemoveArgsAndGlobalTensors();
[all …]
/external/llvm-project/mlir/include/mlir/IR/
DBuiltinOps.h37 class OwningModuleRef : public OwningOpRef<ModuleOp> {
39 using OwningOpRef<ModuleOp>::OwningOpRef;
41 OwningModuleRef(OwningOpRef<ModuleOp> &&other) in OwningModuleRef()
42 : OwningOpRef<ModuleOp>(std::move(other)) {} in OwningModuleRef()
78 struct PointerLikeTypeTraits<mlir::ModuleOp> {
80 static inline void *getAsVoidPointer(mlir::ModuleOp val) {
83 static inline mlir::ModuleOp getFromVoidPointer(void *p) {
84 return mlir::ModuleOp::getFromOpaquePointer(p);
/external/llvm-project/mlir/lib/Dialect/SPIRV/Serialization/
DTranslateRegistration.cpp58 OwningModuleRef module(ModuleOp::create(FileLineColLoc::get( in deserializeModule()
81 static LogicalResult serializeModule(ModuleOp module, raw_ostream &output) { in serializeModule()
87 SmallVector<spirv::ModuleOp, 1> spirvModules; in serializeModule()
88 module.walk([&](spirv::ModuleOp op) { spirvModules.push_back(op); }); in serializeModule()
110 [](ModuleOp module, raw_ostream &output) { in registerToSPIRVTranslation()
123 static LogicalResult roundTripModule(ModuleOp srcModule, bool emitDebugInfo, in roundTripModule()
127 auto spirvModules = srcModule.getOps<spirv::ModuleOp>(); in roundTripModule()
147 OwningModuleRef dstModule(ModuleOp::create(FileLineColLoc::get( in roundTripModule()
159 [](ModuleOp module, raw_ostream &output) { in registerTestRoundtripSPIRV()
170 [](ModuleOp module, raw_ostream &output) { in registerTestRoundtripDebugSPIRV()
/external/llvm-project/mlir/include/mlir/Reducer/Passes/
DOpReducer.h28 llvm::function_ref<std::vector<Operation *>(ModuleOp)> getSpecificOps);
34 std::vector<bool> initTransformSpace(ModuleOp module);
47 llvm::function_ref<void(ModuleOp, int, int)> transform);
50 llvm::function_ref<std::vector<Operation *>(ModuleOp)> getSpecificOps;
62 static std::vector<Operation *> getSpecificOps(ModuleOp module) { in getSpecificOps()
71 static void deleteOps(ModuleOp module, int start, int end) { in deleteOps()
90 std::vector<bool> initTransformSpace(ModuleOp module) { in initTransformSpace()
/external/llvm-project/mlir/include/mlir/Dialect/SPIRV/
DPasses.h21 class ModuleOp; variable
26 std::unique_ptr<OperationPass<mlir::ModuleOp>>
37 std::unique_ptr<OperationPass<spirv::ModuleOp>>
47 std::unique_ptr<OperationPass<spirv::ModuleOp>> createLowerABIAttributesPass();
51 std::unique_ptr<OperationPass<spirv::ModuleOp>> createRewriteInsertsPass();
DModuleCombiner.h24 class ModuleOp; variable
71 combine(llvm::MutableArrayRef<ModuleOp> modules,
73 llvm::function_ref<void(ModuleOp, StringRef, StringRef)>
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dkernel_lowering.h34 Status LowerLHLOToGPU(mlir::ModuleOp module,
37 Status LowerKernelBodiesToNVVM(mlir::ModuleOp module);
39 Status LowerKernelBodiesToROCDL(mlir::ModuleOp module);
41 StatusOr<mlir::ModuleOp> ExtractKernelModule(mlir::ModuleOp module);
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dpasses.h34 std::unique_ptr<OperationPass<ModuleOp>>
40 std::unique_ptr<OperationPass<ModuleOp>> CreateEmbedTFFrameworkAssertPass();
50 std::unique_ptr<OperationPass<ModuleOp>> CreateTFKernelToLLVMPass(
55 std::unique_ptr<OperationPass<ModuleOp>> CreateShapeToDescriptorsPass();
59 std::unique_ptr<OperationPass<ModuleOp>> CreateHloBufferizePass();
63 std::unique_ptr<OperationPass<ModuleOp>> CreateFinalBufferizePass();
/external/llvm-project/mlir/test/lib/Rewrite/
DTestPDLByteCode.cpp48 : public PassWrapper<TestPDLByteCodePass, OperationPass<ModuleOp>> {
50 ModuleOp module = getOperation(); in runOnOperation()
54 ModuleOp patternModule = module.lookupSymbol<ModuleOp>("patterns"); in runOnOperation()
55 ModuleOp irModule = module.lookupSymbol<ModuleOp>("ir"); in runOnOperation()
/external/llvm-project/mlir/include/mlir/Reducer/
DReductionTreeUtils.h31 void updateGoldenModule(ModuleOp &golden, ModuleOp reduced);
40 std::vector<bool> createTransformSpace(ModuleOp module, int numIndices);
46 llvm::function_ref<void(ModuleOp, int, int)> transform,
DReductionNode.h33 ReductionNode(ModuleOp module, ReductionNode *parent);
35 ReductionNode(ModuleOp module, ReductionNode *parent,
42 ModuleOp getModule() const { return module; } in getModule()
76 ModuleOp module;
/external/llvm-project/mlir/lib/Dialect/SPIRV/Linking/ModuleCombiner/
DModuleCombiner.cpp28 spirv::ModuleOp combinedModule) { in renameSymbol()
47 spirv::ModuleOp target, in updateSymbolAndAllUses()
48 spirv::ModuleOp source, in updateSymbolAndAllUses()
123 combine(llvm::MutableArrayRef<spirv::ModuleOp> modules, in combine()
125 llvm::function_ref<void(ModuleOp, StringRef, StringRef)> in combine() argument
135 auto combinedModule = combinedModuleBuilder.create<spirv::ModuleOp>( in combine()
146 DenseMap<StringRef, spirv::ModuleOp> symNameToModuleMap; in combine()
156 spirv::ModuleOp moduleClone = module.clone(); in combine()
175 spirv::ModuleOp originalModule = in combine()
/external/llvm-project/mlir/tools/mlir-reduce/
DReductionTreeUtils.cpp21 void ReductionTreeUtils::updateGoldenModule(ModuleOp &golden, in updateGoldenModule()
22 ModuleOp reduced) { in updateGoldenModule()
49 std::vector<bool> ReductionTreeUtils::createTransformSpace(ModuleOp module, in createTransformSpace()
102 llvm::function_ref<void(ModuleOp, int, int)> transform, bool isDeletion) { in createVariants() argument
104 ModuleOp module = parent->getModule(); in createVariants()
116 ModuleOp variantModule = module.clone(); in createVariants()
131 ModuleOp variantModule = module.clone(); in createVariants()
/external/llvm-project/mlir/tools/mlir-spirv-cpu-runner/
Dmlir-spirv-cpu-runner.cpp48 convertMLIRModule(ModuleOp module, llvm::LLVMContext &context) { in convertMLIRModule()
50 auto modules = module.getOps<ModuleOp>(); in convertMLIRModule()
57 ModuleOp nested = *modules.begin(); in convertMLIRModule()
68 static LogicalResult runMLIRPasses(ModuleOp module) { in runMLIRPasses()
74 OpPassManager &nestedPM = passManager.nest<spirv::ModuleOp>(); in runMLIRPasses()
/external/llvm-project/mlir/unittests/Pass/
DAnalysisManagerTest.cpp28 OpSpecificAnalysis(ModuleOp) {} in OpSpecificAnalysis()
35 OwningModuleRef module(ModuleOp::create(UnknownLoc::get(&context))); in TEST()
57 OwningModuleRef module(ModuleOp::create(UnknownLoc::get(&context))); in TEST()
87 OwningModuleRef module(ModuleOp::create(UnknownLoc::get(&context))); in TEST()
131 OwningModuleRef module(ModuleOp::create(UnknownLoc::get(&context))); in TEST()
153 OwningModuleRef module(ModuleOp::create(UnknownLoc::get(&context))); in TEST()
158 am.getAnalysis<OpSpecificAnalysis, ModuleOp>(); in TEST()
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dtfr_decompose_ctx.h43 explicit TFRDecomposeContext(mlir::ModuleOp tfr_module);
57 Status DecomposeGraph(mlir::ModuleOp user_module);
63 mlir::ModuleOp tfr_module_;
76 Status DecomposeGraph(mlir::ModuleOp user_module);
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dpasses.h29 class ModuleOp; variable
54 std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTfTypesPass();
69 std::unique_ptr<OperationPass<ModuleOp>> createLegalizeTFControlFlowPass();
86 std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTFCommunicationPass();
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.h71 mlir::ModuleOp module_op, llvm::StringRef device_type,
86 mlir::ModuleOp module);
91 Status BuildHloFromTf(mlir::ModuleOp module_op, xla::XlaBuilder& builder,
103 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes,
112 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes,
136 mlir::ModuleOp module_op, llvm::ArrayRef<XlaArgument> args,
Dtranslate_utils.h29 void PopulateTfVersions(mlir::ModuleOp module, const VersionDef& versions);
34 mlir::LogicalResult ExtractTfVersions(mlir::ModuleOp module,
40 mlir::ModuleOp module);
/external/llvm-project/mlir/lib/Conversion/GPUToVulkan/
DConvertGPULaunchFuncToVulkanLaunchFunc.cpp49 LogicalResult createBinaryShader(ModuleOp module,
96 llvm::make_early_inc_range(getOperation().getOps<spirv::ModuleOp>())) in runOnOperation()
132 ModuleOp module, std::vector<char> &binaryShader) { in createBinaryShader()
135 for (auto spirvModule : module.getOps<spirv::ModuleOp>()) { in createBinaryShader()
151 ModuleOp module = getOperation(); in convertGpuLaunchFunc()
190 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
/external/llvm-project/mlir/include/mlir/Conversion/SPIRVToLLVM/
DConvertSPIRVToLLVMPass.h19 class ModuleOp; variable
31 std::unique_ptr<OperationPass<ModuleOp>> createLowerHostCodeToLLVMPass();
34 std::unique_ptr<OperationPass<ModuleOp>> createConvertSPIRVToLLVMPass();
/external/llvm-project/mlir/include/mlir/Conversion/GPUToVulkan/
DConvertGPUToVulkanPass.h23 class ModuleOp; variable
26 std::unique_ptr<OperationPass<ModuleOp>>
29 std::unique_ptr<OperationPass<mlir::ModuleOp>>

12345678910>>...14