- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for clustering_passes (0.27 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/clustering_passes.td
Mason Chang <******@****.***> 1714440931 -0700
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 19.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/BUILD
"//tensorflow/compiler/mlir/tf2xla/internal:__subpackages__", ], licenses = ["notice"], ) cc_library( name = "clustering_passes", hdrs = [ "clustering_passes.h", ], textual_hdrs = [ "clustering_passes.h.inc", ], deps = [ ":extract_head_tail_outside_compilation", ":extract_outside_compilation", ":hoist_broadcast_read",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 17:50:14 UTC 2024 - 17.9K 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) -
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h" #include "tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/sparsecore_passes.h" #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h" namespace tensorflow { namespace tf2xla { namespace internal { using mlir::OpPassManager; using mlir::func::FuncOp; // Adds replicated Bridge clustering pipeline passes to the given pass_manager.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 16:09:14 UTC 2024 - 11.2K bytes - Viewed (1) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
using mlir::tf_device::LaunchOp; using mlir::tf_device::ParallelExecuteOp; using mlir::tf_device::ReplicateOp; #define GEN_PASS_DEF_XLABROADCASTPASS #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc" const char kICIWeightDistributionMlirBridgeMarker[] = "ici_weight_distribution_mlir_bridge_marker"; struct XlaBroadcast : public impl::XlaBroadcastPassBase<XlaBroadcast> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0)