| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/ |
| H A D | test_passes.cc | 92 void runOnOperation() final { in runOnOperation() function in mlir::gml_st::__anon085edece0111::TestGmlStLoopPeelingPass
148 void runOnOperation() override { in runOnOperation() function
175 void runOnOperation() override { in runOnOperation() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/graph_compactor/ |
| H A D | pass.cc | 90 void runOnOperation() override { in runOnOperation() function in mlir::tfg::__anon411976f40111::NameCompressPass
179 void runOnOperation() override { in runOnOperation() function in mlir::tfg::__anon411976f40411::StripDefaultAttrsPass
266 void runOnOperation() override { in runOnOperation() function in mlir::tfg::__anon411976f40611::AddDefaultAttrsPass
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/ |
| H A D | bufferize_pass.cc | 139 void runOnOperation() override { in runOnOperation() function
237 void runOnOperation() override { in runOnOperation() function
283 void runOnOperation() override { in runOnOperation() function
|
| H A D | gpu_kernel_lowering_passes.cc | 63 void GpuKernelToNVVMPass::runOnOperation() { in runOnOperation() function in GpuKernelToNVVMPass
77 void GpuKernelToROCDLPass::runOnOperation() { in runOnOperation() function in GpuKernelToROCDLPass
|
| H A D | buffer_packing.cc | 430 void runOnOperation() override { in runOnOperation() function
445 void runOnOperation() override { in runOnOperation() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/ |
| H A D | dtensor_embedding_v2_stub.cc | 24 void runOnOperation() override {} in runOnOperation() function
36 void runOnOperation() override {} in runOnOperation() function
|
| H A D | collective_lowering.cc | 732 void runOnOperation() override { in runOnOperation() function
755 void runOnOperation() override { in runOnOperation() function
773 void runOnOperation() override { in runOnOperation() function
790 void runOnOperation() override { in runOnOperation() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/consolidate_attrs/ |
| H A D | pass.cc | 341 void ConsolidateAttributesPassImpl::runOnOperation() { in runOnOperation() function in mlir::tfg::ConsolidateAttributesPassImpl
635 void PrepareAttributesForExportPassImpl::runOnOperation() { in runOnOperation() function in mlir::tfg::PrepareAttributesForExportPassImpl
666 void runOnOperation() override { in runOnOperation() function
679 void runOnOperation() override { in runOnOperation() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | lift_variables_test_pass.cc | 36 void runOnOperation() override { in runOnOperation() function in mlir::tf_saved_model::__anon3cdf97880111::LiftVariablesTestPass
51 void runOnOperation() override { in runOnOperation() function in mlir::tf_saved_model::__anon3cdf97880111::LiftVariablesInvalidSessionTestPass
|
| H A D | mark_initialized_variables_test_pass.cc | 40 void runOnOperation() override { in runOnOperation() function in mlir::__anon7d1b25b90111::MarkInitializedVariablesTestPass
64 void runOnOperation() override { in runOnOperation() function in mlir::__anon7d1b25b90111::MarkInitializedVariablesInvalidSessionTestPass
|
| H A D | drop_while_shape_invariant.cc | 44 void DropWhileShapeInvariantPass::runOnOperation() { in runOnOperation() function in mlir::TF::__anon094f88f40111::DropWhileShapeInvariantPass
48 void DropWhileShapeInvariantInDeviceClusterPass::runOnOperation() { in runOnOperation() function in mlir::TF::__anon094f88f40111::DropWhileShapeInvariantInDeviceClusterPass
|
| H A D | tf_device_assignment.cc | 39 void runOnOperation() override { in runOnOperation() function in mlir::TF::__anon5e648b0d0111::SimpleTFDeviceAssignmentPass
67 void runOnOperation() override { in runOnOperation() function in mlir::TF::__anon5e648b0d0111::TFDeviceAssignmentByFuncAttrPass
|
| H A D | init_text_file_to_import_test_pass.cc | 59 void InitTextFileToImportTestPass::runOnOperation() { in runOnOperation() function in mlir::TF::__anon029da5b40111::InitTextFileToImportTestPass
117 void InitTextFileToImportSavedModelTestPass::runOnOperation() { in runOnOperation() function in mlir::TF::__anon029da5b40111::InitTextFileToImportSavedModelTestPass
|
| H A D | lift_variables_test_pass.h | 126 void runOnOperation() override { in runOnOperation() function
140 void runOnOperation() override { in runOnOperation() function
|
| H A D | decompose_resource_ops_pass.cc | 155 void runOnOperation() override { in runOnOperation() function
171 void runOnOperation() override { in runOnOperation() function
|
| H A D | tf_graph_optimization_pass.cc | 64 void GraphOptPass::runOnOperation() { in runOnOperation() function in tensorflow::GraphOptPass
171 void runOnOperation() override { in runOnOperation() function in tensorflow::GraphOptByNamePass
|
| H A D | cluster_outlining.cc | 147 void ClusterOutliningPass::runOnOperation() { in runOnOperation() function in mlir::TFDevice::__anon10ec6c260111::ClusterOutliningPass
156 void LaunchOutliningPass::runOnOperation() { in runOnOperation() function in mlir::TFDevice::__anon10ec6c260111::LaunchOutliningPass
|
| H A D | promote_resources_to_args.cc | 357 void PromoteResourcesToArgsPass::runOnOperation() { in runOnOperation() function in mlir::TF::__anon60b3149a0111::PromoteResourcesToArgsPass
387 void PromoteVarHandlesToArgsPass::runOnOperation() { in runOnOperation() function in mlir::TF::__anon60b3149a0111::PromoteVarHandlesToArgsPass
|
| H A D | layout_optimization.cc | 103 void LayoutAssignmentPass::runOnOperation() { in runOnOperation() function in mlir::TF::__anon0e7f28700111::LayoutAssignmentPass
435 void MoveTransposesPass::runOnOperation() { in runOnOperation() function in mlir::TF::__anon0e7f28700111::MoveTransposesPass
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| H A D | tf_jitrt_test_passes.cc | 45 void runOnOperation() override { in runOnOperation() function
68 void runOnOperation() override { in runOnOperation() function
|
| H A D | tf_jitrt_tile_cwise.cc | 138 void runOnOperation() override { in runOnOperation() function
148 void runOnOperation() override { in runOnOperation() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/metrics/ |
| H A D | error_collector_inst_test.cc | 58 void runOnOperation() override { in runOnOperation() function in mlir::TFL::__anon4eb5ef7b0111::MockSuccessPass
79 void runOnOperation() override { in runOnOperation() function in mlir::TFL::__anon4eb5ef7b0111::MockFailurePass
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | xla_legalize_tf.cc | 346 void LegalizeTF::runOnOperation() { in runOnOperation() function in mlir::mhlo::__anonbcb590680111::LegalizeTF
358 void LegalizeTFModulePass::runOnOperation() { in runOnOperation() function in mlir::mhlo::__anonbcb590680111::LegalizeTFModulePass
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | tfg_optimizer_hook_test.cc | 47 void runOnOperation() override { in runOnOperation() function in mlir::tfg::__anon57fdb71a0111::TestPass
60 void runOnOperation() override { signalPassFailure(); } in runOnOperation() function in mlir::tfg::__anon57fdb71a0111::AlwaysFailPass
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | post_quantize.cc | 367 void PostQuantizePass::runOnOperation() { in runOnOperation() function in mlir::TFL::__anonfa26bdbd0111::PostQuantizePass
399 void PostQuantizeRemoveQDQPass::runOnOperation() { in runOnOperation() function in mlir::TFL::__anonfa26bdbd0111::PostQuantizeRemoveQDQPass
|