- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for TPUCompilationResult (0.3 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/embedding_sequencing.mlir
// CHECK-DAG: TPUReplicateMetadata // CHECK-DAG: TPUCompilationResult // CHECK-DAG: TPUReplicatedOutput // CHECK: return // core_tput should have TPU ops including both replicated inputs and outputs // CHECK: func.func private @_func_core_tpu // CHECK-DAG: TPUReplicatedInput // CHECK-DAG: TPUReplicateMetadata // CHECK-DAG: TPUCompilationResult // CHECK-DAG: TPUReplicatedOutput // CHECK: return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 01 21:27:49 UTC 2023 - 19.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/embedding_pipelining.mlir
"tf.TPUReplicateMetadata"() {_has_manual_control_dependencies = true, _replication_info = "repl_info", num_replicas = 1 : i64} : () -> () // expected-error @+1 {{'tf.TPUCompilationResult' op is not part of the replication region "repl_info" vs "wrong_repl_info"}} %1 = "tf.TPUCompilationResult"() {_tpu_compilation_status = "wrong_repl_info"} : () -> tensor<!tf_type.string> return %arg0 : tensor<i1> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 33.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
// all graphs should have control dependencies to enforce this but this is // currently not the case (see b/177478741). const llvm::SmallVector<std::string, 4> ops_to_preserve = { "tf.TPUReplicateMetadata", "tf.TPUCompilationResult", "tf.TPUReplicatedOutput"}; bool strict_clusters = tensorflow::GetMlirCommonFlags()->tf_mlir_enable_strict_clusters; pm.addNestedPass<FuncOp>(
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/tensorflow/transforms/sparsecore/embedding_sequencing.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/testdata/outside_compilation.mlir
%outputs_25, %control_26 = tf_executor.island(%control_18) wraps "tf.TPUCompilationResult"() {_tpu_compilation_status = "cluster_sample_sequence", device = ""} : () -> tensor<!tf_type.string>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 19 20:19:45 UTC 2023 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
// returns `std::nullopt`. std::optional<llvm::StringRef> GetTpuClusterName(Operation* op) { if (auto tpu_status = op->getAttrOfType<StringAttr>(kTpuStatusAttr)) { // Borrow cluster name from TPU status (for `TPUCompilationResult` op). return tpu_status.getValue(); } auto device_type = op->getAttrOfType<StringAttr>(TF::kCompileDeviceTypeAttr); if (!device_type || device_type.getValue() != TF::kTpuDevice) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_sharding_identification.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 20 19:07:52 UTC 2024 - 47.5K bytes - Viewed (0)