- Sort Score
- Result 10 results
- Languages All
Results 11 - 15 of 15 for TPUCompilationResult (0.3 sec)
-
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_rewrite.mlir
%compile_result = "tf.TPUCompilationResult"() {_tpu_compilation_status = "cluster0"} : () -> tensor<!tf_type.string> %compile_result2 = "tf.TPUCompilationResult"() {_tpu_compilation_status = "cluster0"} : () -> tensor<!tf_type.string> // CHECK-NOT: "tf.TPUCompilationResult" // CHECK: return %[[EXECUTE_OUTPUT]], %[[COMPILE_OUTPUT]]#0, %[[COMPILE_OUTPUT]]#0
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K 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) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); TF_DerivedOperandTypeListAttr Tin = TF_DerivedOperandTypeListAttr<0>; TF_DerivedResultTypeListAttr Tout = TF_DerivedResultTypeListAttr<0>; } def TF_TPUCompilationResultOp : TF_Op<"TPUCompilationResult", [TF_MustExecute]> { let summary = "Returns the result of a TPU compilation."; let description = [{ This operation returns the result of a TPU compilation as a serialized
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)