| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | passes.h | 31 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 …]
|
| D | test_passes.h | 28 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>>
|
| D | bridge.h | 31 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);
|
| D | tf_saved_model_passes.h | 30 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>>
|
| D | lift_variables_test_pass.cc | 37 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/ |
| D | create_dtensor_mlir_passes.h | 66 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 …]
|
| D | Passes.td | 118 : 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/ |
| D | passes.h | 46 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/ |
| D | dump_mlir_util_test.cc | 32 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 …]
|
| D | device_util_test.cc | 67 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/ |
| D | import_model.h | 43 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);
|
| D | tf_mlir_translate.h | 42 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/mlir/tfrt/transforms/ |
| D | passes.h | 43 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/xla/mlir_hlo/include/mlir-hlo/Transforms/ |
| D | passes.h | 25 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/quantization/tensorflow/passes/ |
| D | passes.h | 32 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/ |
| D | passes.h | 36 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/core/ir/ |
| D | ops_test.cc | 31 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()
|
| D | tf_op_registry_test.cc | 54 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()
|
| D | utility_test.cc | 40 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/lite/experimental/tac/transforms/ |
| D | passes.h | 41 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/compiler/mlir/xla/ |
| D | xla_mlir_translate.cc | 54 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/ |
| D | passes.h | 33 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/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| D | passes.h | 26 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/mlir/tfr/integration/ |
| D | tfr_decompose_ctx.h | 44 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/pjrt/ |
| D | mlir_to_hlo.cc | 34 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()
|