Home
last modified time | relevance | path

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

12345678910>>...18

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
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 …]
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>>
Dbridge.h31 tensorflow::Status TPUBridge(ModuleOp module, bool enable_logging,
40 tensorflow::Status TPUBridgeV1Compat(ModuleOp module, bool enable_logging,
51 tensorflow::Status RunBridgeWithStandardPipeline(ModuleOp module,
56 tensorflow::Status RunTFXLABridge(ModuleOp module, bool enable_logging);
Dtf_saved_model_passes.h30 std::unique_ptr<OperationPass<ModuleOp>> CreateOptimizeGlobalTensorsPass();
33 std::unique_ptr<OperationPass<ModuleOp>> CreateFreezeGlobalTensorsPass(
37 std::unique_ptr<OperationPass<ModuleOp>> CreateFreezeAssetsPass(
47 std::unique_ptr<OperationPass<ModuleOp>>
Dlift_variables_test_pass.cc37 ModuleOp module = getOperation(); in runOnOperation()
52 ModuleOp module = getOperation(); in runOnOperation()
64 std::unique_ptr<OperationPass<ModuleOp>> CreateLiftVariablesTestPass() { in CreateLiftVariablesTestPass()
68 std::unique_ptr<OperationPass<ModuleOp>>
/external/tensorflow/tensorflow/dtensor/mlir/
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 …]
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 …]
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
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 …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
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 …]
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 …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
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);
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>>
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Transforms/
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();
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
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>>
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
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(
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
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();
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
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(
/external/tensorflow/tensorflow/core/ir/
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()
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()
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()
/external/tensorflow/tensorflow/compiler/mlir/xla/
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()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dpasses.h33 class ModuleOp; variable
54 std::unique_ptr<OperationPass<ModuleOp>> createLegalizeTFModulePass(
90 std::unique_ptr<OperationPass<ModuleOp>> createLegalizeTFControlFlowPass();
108 std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTFCommunicationPass();
112 std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTFCollectivePass();
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dtfr_decompose_ctx.h44 explicit TFRDecomposeContext(mlir::ModuleOp tfr_module);
58 Status DecomposeGraph(mlir::ModuleOp user_module);
64 mlir::ModuleOp tfr_module_;
77 Status DecomposeGraph(mlir::ModuleOp user_module);
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/transforms/
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();
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dmlir_to_hlo.cc34 Status MlirToXlaComputation(mlir::ModuleOp module, in MlirToXlaComputation()
71 StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> ParseMlirModuleString( in ParseMlirModuleString()
73 mlir::OwningOpRef<mlir::ModuleOp> module; in ParseMlirModuleString()
78 module = mlir::parseSourceString<mlir::ModuleOp>( in ParseMlirModuleString()
96 TF_ASSIGN_OR_RETURN(mlir::OwningOpRef<mlir::ModuleOp> module, in ParseMlirModuleStringAndConvertToXlaComputation()

12345678910>>...18