Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 13 of 13 for TPUCompilationResult (0.44 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc

                                   llvm::SetVector<Operation*>& merged_set,
                                   OpType& found_op) {
      // Find the TPUReplicationMetadata or TPUCompilationResult ops which will be
      // cloned/inserted into each region. We add them to the merged_set so that
      // they're ignored when extracting the four main functions.
      found_op = nullptr;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 92.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/tpu_sharding_identification.mlir

      %cst = "tf.Const"() {value = dense<5.000000e+00> : tensor<f32>} : () -> tensor<f32>
      %0 = "tf.TPUCompilationResult"() {_tpu_compilation_status = "cluster_tpu_function", device = ""} : () -> tensor<!tf_type.string>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Feb 20 19:07:52 UTC 2024
    - 47.5K bytes
    - Viewed (0)
  3. 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)
Back to top