Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 34 of 34 for BatchMatMulV2 (0.25 sec)

  1. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

      let summary = "Batch Matrix Multiply Operator";
    
      let description = [{
    Performs a batched matrix multiplication on the inputs. Follows the
    conventions of TensorFlow BatchMatMulV2, with support for unknown dimensions
    in the batch dimensions and broadcasting.
    
        Inputs:
          `inputs[0]`: required: input LHS
          `inputs[1]`: required: input RHS
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

                                               Value *out_lhs, Value *out_rhs,
                                               PatternRewriter *rewriter) {
      // The dimension structure of the relevant operands to a tf.BatchMatMulV2 is:
      // - lhs: [LHSBATCHDIMS..., LHSROWS, LHSCOLS]
      // - rhs: [RHSBATCHDIMS..., RHSROWS, RHSCOLS]
      // - result: [broadcast(LHSBATCHDIMS, RHSBATCHDIMS)..., LHSROWS, RHSCOLS]
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      );
    
      TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
    
      let hasCanonicalizer = 1;
    
      let hasVerifier = 1;
    }
    
    def TF_BatchMatMulV2Op : TF_Op<"BatchMatMulV2", [Pure, TF_SameOperandsAndResultElementTypeResolveRef]> {
      let summary = "Multiplies slices of two tensors in batches.";
    
      let description = [{
    Multiplies all slices of `Tensor` `x` and `y` (each slice can be
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
  4. RELEASE.md

        *   Remove unused `StringViewVariantWrapper`.
        *   Delete unused `Fingerprint64Map` op registration
        *   Add broadcasting support to `tf.matmul`.
        *   Add C++ Gradient for `BatchMatMulV2`.
        *   Add `tf.math.cumulative_logsumexp` operation.
        *   Add ellipsis (...) support for `tf.einsum()`.
        *   Add expand_composites argument to all `nest.*` methods.
        *   Added `strings.byte_split`.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 730.3K bytes
    - Viewed (0)
Back to top