Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for RegisterTPULowerClusterToRuntimeOpsPassPipeline (0.33 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops.h

        llvm::StringRef module_name = llvm::StringRef());
    
    // The same API as RunLowerClusterToRuntimeOpsPassPipeline but as an MLIR pass
    // pipeline.
    void RegisterTPULowerClusterToRuntimeOpsPassPipeline();
    void RegisterNonTPULowerClusterToRuntimeOpsPassPipeline();
    
    }  // namespace tfrt_compiler
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 21:47:17 UTC 2023
    - 2.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf_mlir_opt_main.cc

      tensorflow::RegisterMlProgramPasses();
      mlir::TFTPU::registerRuntimeLoweringPasses();
      mlir::TFDevice::registerSparseCorePasses();
    
      tensorflow::tfrt_compiler::RegisterTPULowerClusterToRuntimeOpsPassPipeline();
      tensorflow::tfrt_compiler::
          RegisterNonTPULowerClusterToRuntimeOpsPassPipeline();
    
      mlir::DialectRegistry registry;
      mlir::RegisterCommonToolingDialects(registry);
    
      return failed(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:16:49 UTC 2024
    - 4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops.cc

      return absl::OkStatus();
    }
    
    // TODO(b/305211853): Unify the CPU/TPU/GPU Execution Ops and thus these two
    // passes should merge together.
    void RegisterTPULowerClusterToRuntimeOpsPassPipeline() {
      static mlir::PassPipelineRegistration<StandardPipelineOptions> pipeline(
          "tfrt-lower-cluster-to-runtime-ops-tpu",
          "Run all the passes involved after the clustering transformations from "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 17 18:52:57 UTC 2024
    - 9.4K bytes
    - Viewed (0)
Back to top