- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for clustering_passes (0.19 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h
#define GEN_PASS_DECL_TPUSHARDINGIDENTIFICATIONPASS #define GEN_PASS_DECL_VERIFYCLUSTERINGPASS #define GEN_PASS_DECL_XLACLUSTERFORMATIONPASS #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc" } // namespace internal } // namespace tf2xla } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_clustering_pass_test.cc
#include "mlir/IR/OwningOpRef.h" // from @llvm-project #include "mlir/Pass/PassManager.h" // from @llvm-project #include "mlir/Support/LogicalResult.h" // from @llvm-project #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h" #include "tensorflow/compiler/mlir/tf2xla/transforms/test_utils.h" #include "tsl/platform/statusor.h" namespace tensorflow { namespace tf2xla { namespace internal { namespace {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 17:03:53 UTC 2023 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_clustering_pass.cc
namespace tensorflow { namespace tf2xla { namespace internal { namespace { #define GEN_PASS_DEF_VERIFYCLUSTERINGPASS #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc" using mlir::Operation; using mlir::WalkResult; class VerifyClusteringPass : public impl::VerifyClusteringPassBase<VerifyClusteringPass> { public: void runOnOperation() override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 20:33:37 UTC 2023 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf_mlir_opt_main.cc
#include "tensorflow/compiler/mlir/tensorflow/utils/mlprogram_util.h" #include "tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h" #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h" #include "tensorflow/compiler/mlir/tf2xla/internal/passes/mlir_to_graph_passes.h" #include "tensorflow/compiler/mlir/tf2xla/transforms/passes.h" #include "tensorflow/compiler/mlir/tosa/tf_passes.h"
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/tf2xla/internal/passes/hoist_broadcast_read.cc
using mlir::WalkResult; using mlir::func::FuncOp; using mlir::TF::ReadVariableOp; using mlir::tf_device::ReplicateOp; #define GEN_PASS_DEF_HOISTBROADCASTREADPASS #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc" constexpr char kFuncDeviceAttr[] = "tf.device"; constexpr char kCpuDeviceType[] = "CPU"; struct HoistBroadcastRead : public impl::HoistBroadcastReadPassBase<HoistBroadcastRead> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_cluster_formation.cc
using mlir::SymbolTableCollection; using mlir::SymbolUserOpInterface; using mlir::func::FuncOp; #define GEN_PASS_DEF_XLACLUSTERFORMATIONPASS #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc" // Outlines partitioned call ops with `_XlaMustCompile` to device clusters. struct XlaClusterFormationPass : public impl::XlaClusterFormationPassBase<XlaClusterFormationPass> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 6K bytes - Viewed (0)