Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for DiagPart (0.38 sec)

  1. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // CHECK: return [[CLAMP]]
      func.return %0 : tensor<?xf32>
    }
    
    //===----------------------------------------------------------------------===//
    // DiagPart
    //===----------------------------------------------------------------------===//
    
    // -----
    
    // CHECK-LABEL: func @diag_part
    // CHECK-SAME: %[[ARG:.*]]: tensor<4x3x4x3xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      );
    
      TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
    }
    
    def TF_DiagPartOp : TF_Op<"DiagPart", [Pure]> {
      let summary = "Returns the diagonal part of the tensor.";
    
      let description = [{
    This operation returns a tensor with the `diagonal` part
    of the `input`. The `diagonal` part is computed as follows:
    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