Home
last modified time | relevance | path

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

12345678910

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dpasses.h30 std::unique_ptr<OperationPass<ModuleOp>> CreateBreakUpIslandsPass();
63 std::unique_ptr<OperationPass<ModuleOp>>
68 std::unique_ptr<OperationPass<ModuleOp>>
76 std::unique_ptr<OperationPass<ModuleOp>> CreateTFShapeInferencePass();
82 std::unique_ptr<OperationPass<ModuleOp>> CreateGuaranteeAllFuncsOneUsePass();
143 std::unique_ptr<OperationPass<ModuleOp>> CreateResourceDeviceInferencePass();
150 std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteResourcesToArgsPass();
154 std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteVarHandlesToArgsPass();
172 std::unique_ptr<OperationPass<ModuleOp>> CreateStackOpsDecompositionPass();
176 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(
35 std::unique_ptr<OperationPass<ModuleOp>> CreateFreezeAssetsPass(
45 std::unique_ptr<OperationPass<ModuleOp>>
51 std::unique_ptr<OperationPass<ModuleOp>> CreateLiftVariablesPass(
63 std::unique_ptr<OperationPass<ModuleOp>>
67 std::unique_ptr<OperationPass<ModuleOp>> CreateFreezeVariablesPass(
Doptimize_global_tensors.cc44 : public PassWrapper<OptimizeGlobalTensorsPass, OperationPass<ModuleOp>> {
90 GlobalTensorUsesMap CreateGlobalTensorUsesMap(ModuleOp module) { in CreateGlobalTensorUsesMap()
116 ModuleOp module, const GlobalTensorUsesMap& global_tensor_uses_map, in MarkGlobalTensorsImmutable()
127 void EraseUnusedGlobalTensors(ModuleOp module, in EraseUnusedGlobalTensors()
144 void EraseUnusedBoundInputs(ModuleOp module) { in EraseUnusedBoundInputs()
179 std::unique_ptr<OperationPass<ModuleOp>> CreateOptimizeGlobalTensorsPass() { in CreateOptimizeGlobalTensorsPass()
Dlift_variables_test_pass.cc28 : public PassWrapper<LiftVariablesTestPass, OperationPass<ModuleOp>> {
43 ModuleOp module = getOperation(); in runOnOperation()
55 OperationPass<ModuleOp>> {
67 ModuleOp module = getOperation(); in runOnOperation()
Dbridge.h28 tensorflow::Status TPUBridge(ModuleOp module, bool enable_logging);
34 tensorflow::Status TPUBridgeV1Compat(ModuleOp module, bool enable_logging);
44 tensorflow::Status RunBridgeWithStandardPipeline(ModuleOp module,
Dinit_text_file_to_import_test_pass.cc39 OperationPass<ModuleOp>> {
56 ModuleOp module = getOperation(); in runOnOperation()
104 OperationPass<ModuleOp>> {
121 ModuleOp module = getOperation(); in runOnOperation()
Dinitialize_variables_in_session_init.cc38 OperationPass<ModuleOp>> {
85 FuncOp CreateSessionInitFunc(ModuleOp module) { in CreateSessionInitFunc()
113 FuncOp GetOrCreateSessionInitFunc(ModuleOp module) { in GetOrCreateSessionInitFunc()
127 ModuleOp module = getOperation(); in runOnOperation()
185 std::unique_ptr<OperationPass<ModuleOp>>
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dpasses.h26 class ModuleOp; variable
52 std::unique_ptr<OperationPass<ModuleOp>> CreateLowerStaticTensorListPass(
79 std::unique_ptr<OperationPass<ModuleOp>> CreateTrimFunctionsPass(
84 std::unique_ptr<OperationPass<ModuleOp>> CreatePrepareCompositeFunctionsPass();
90 std::unique_ptr<OperationPass<ModuleOp>> CreateOptimizeFunctionalOpsPass();
109 std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTFWhilePass();
112 std::unique_ptr<OperationPass<ModuleOp>> CreateWhileOutlinePass();
126 std::unique_ptr<OperationPass<ModuleOp>>
131 std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeVariablesPass();
135 std::unique_ptr<OperationPass<ModuleOp>> CreateAnalyzeVariablesPass();
[all …]
Dinsert_call_once_op.cc29 OperationPass<ModuleOp>> {
50 ModuleOp module = getOperation(); in runOnOperation()
83 std::unique_ptr<OperationPass<ModuleOp>>
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dpasses.h46 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
52 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
57 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
62 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>> CreateMergeTfIfOpsPass();
66 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
75 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
80 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
93 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
98 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
216 std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
Dremote_run_encapsulate.cc55 OperationPass<ModuleOp>> {
73 ModuleOp module = entry_func->getParentOfType<ModuleOp>(); in EncapsulateFuncAndSerialize()
79 ModuleOp::create(mlir::UnknownLoc::get(entry_func.getContext())); in EncapsulateFuncAndSerialize()
119 ModuleOp module = getOperation(); in runOnOperation()
236 std::unique_ptr<OperationPass<ModuleOp>> CreateDistRemoteRunEncapsulatePass() { in CreateDistRemoteRunEncapsulatePass()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dpasses.h35 std::unique_ptr<OperationPass<ModuleOp>> CreateEmbedTFFrameworkPass();
40 std::unique_ptr<OperationPass<ModuleOp>> CreateRewriteTFFrameworkAssert();
58 std::unique_ptr<OperationPass<ModuleOp>> CreateTFKernelToLLVMPass(
63 std::unique_ptr<OperationPass<ModuleOp>> CreateShapeToDescriptorsPass();
67 std::unique_ptr<OperationPass<ModuleOp>> CreateComputeOpAndFuncBufferizePass();
71 std::unique_ptr<OperationPass<ModuleOp>> CreateFinalBufferizePass();
74 std::unique_ptr<OperationPass<ModuleOp>> CreateConvertToSignlessPass();
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Dpasses.h40 std::unique_ptr<OperationPass<ModuleOp>> CreateRaiseTargetSubgraphsPass();
43 std::unique_ptr<OperationPass<ModuleOp>> CreateAlternativeSubgraphPass(
47 std::unique_ptr<OperationPass<ModuleOp>> CreateComputeCostPass();
50 std::unique_ptr<OperationPass<ModuleOp>> CreatePickSubgraphsPass();
59 std::unique_ptr<OperationPass<ModuleOp>> CreateFoldConstantsToSubgraphPass(
Draise_target_subgraphs.cc101 mlir::OperationPass<ModuleOp>> {
106 ModuleOp module);
109 ModuleOp module);
111 FuncOp BuildFuncOp(Subgraph* subgraph, OpBuilder* builder, ModuleOp module_op,
192 Subgraph* subgraph, OpBuilder* builder, ModuleOp module_op, in BuildFuncOp()
280 ModuleOp module) { in ExtractSubgraphToFunc()
331 ModuleOp module) { in RaiseTargetSubgraphsForBlock()
375 std::unique_ptr<OperationPass<ModuleOp>> CreateRaiseTargetSubgraphsPass() { in CreateRaiseTargetSubgraphsPass()
Dget_alternative_subgraph.cc92 mlir::OperationPass<ModuleOp>> {
108 ModuleOp module, OpBuilder* builder);
121 const InferenceDeviceType& target_device_inference_type, ModuleOp module,
135 ArrayRef<std::string> devices, FuncOp func, ModuleOp module, in GetAlternativeGraphForFunc()
208 const InferenceDeviceType& target_device_inference_type, ModuleOp module, in GetAlternativeViewForSpec()
287 std::unique_ptr<OperationPass<ModuleOp>> CreateAlternativeSubgraphPass( in CreateAlternativeSubgraphPass()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/
Dpasses.h27 class ModuleOp; variable
56 std::unique_ptr<OperationPass<ModuleOp>> createLegalizeToLhloPass();
66 std::unique_ptr<OperationPass<ModuleOp>> createMarkShapeCalcOpPass();
101 std::unique_ptr<OperationPass<ModuleOp>> CreateExpandHloTuplesPass(
153 std::unique_ptr<OperationPass<ModuleOp>> createRalInjectExecutionContextPass(
160 std::unique_ptr<OperationPass<ModuleOp>> createRalToLLVMPass();
/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/tensorflow/utils/
Dcompile_mlir_util.h75 mlir::ModuleOp module_op, llvm::StringRef device_type,
90 mlir::ModuleOp module);
95 Status BuildHloFromTf(mlir::ModuleOp module_op, xla::XlaBuilder& builder,
107 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes,
118 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes,
142 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);
Ddevice_util_test.cc64 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
105 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
114 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
123 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
134 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
145 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
159 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
179 mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); in TEST()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dxla_mlir_translate.cc71 mlir::ModuleOp::create(mlir::UnknownLoc::get(context)); in HloToMlirHloTranslateFunction()
96 mlir::ModuleOp::create(mlir::UnknownLoc::get(context)); in HloTextToMlirHloTranslateFunction()
108 mlir::ModuleOp module, llvm::raw_ostream& output) { in MlirHloToHloTranslateFunction()
134 Status ConvertMlirHloToHloViaBuilder(mlir::ModuleOp module, in ConvertMlirHloToHloViaBuilder()
171 mlir::ModuleOp module, llvm::raw_ostream& output, bool with_layouts, in MlirHloToHloTextTranslateFunctionImpl()
215 mlir::ModuleOp module, llvm::raw_ostream& output) { in MlirHloToHloTextTranslateFunction()
222 mlir::ModuleOp module, llvm::raw_ostream& output) { in MlirHloToHloTextWithLayoutsTranslateFunction()
231 mlir::ModuleOp module, llvm::raw_ostream& output) { in MlirHloToHloTextViaBuilderTranslateFunction()
Dhlo_to_mlir_hlo.h23 class ModuleOp; variable
33 Status ConvertHloToMlirHlo(mlir::ModuleOp module,
40 Status ConvertHloToMlirHlo(mlir::ModuleOp module, xla::HloModule* hlo_module,
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Dpasses.h35 llvm::Optional<ModuleOp> tfr_module = llvm::None);
40 std::unique_ptr<OperationPass<ModuleOp>> CreateRewriteQuantizedIOPass();
44 llvm::Optional<ModuleOp> tfr_module = llvm::None,
/external/tensorflow/tensorflow/compiler/tf2xla/
Dmlir_bridge_pass.cc41 bool HasTPUDevice(mlir::ModuleOp module) { in HasTPUDevice()
51 bool HasTPUOp(mlir::ModuleOp module) { in HasTPUOp()
63 bool HasTPUDevicesAndOps(mlir::ModuleOp module) { in HasTPUDevicesAndOps()
121 mlir::ModuleOp module, const Graph& graph, in Run()
176 mlir::ModuleOp module) { in Run()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.h34 ConvertMlirToGraphdef(mlir::ModuleOp module, const GraphExportConfig& configs);
40 Status ConvertMlirToGraph(mlir::ModuleOp module,
49 Status ConvertMlirToGraph(mlir::ModuleOp module,

12345678910