- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for RegisterTPULowerClusterToRuntimeOpsPassPipeline (0.67 sec)
-
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) -
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) -
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)