Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 136 for mat_mul (0.24 sec)

  1. tensorflow/compiler/aot/tests/tfcompile_test.cc

        matmul.arg0(1, 0) = 4;
        matmul.arg0(1, 1) = 5;
        matmul.arg0(1, 2) = 6;
    
        matmul.arg1(0, 0) = 7;
        matmul.arg1(0, 1) = 8;
        matmul.arg1(1, 0) = 9;
        matmul.arg1(1, 1) = 10;
        matmul.arg1(2, 0) = 11;
        matmul.arg1(2, 1) = 12;
    
        EXPECT_TRUE(matmul.Run());
        EXPECT_EQ(matmul.error_msg(), "");
        const float results[4] = {58, 64, 139, 154};
        for (int i = 0; i < 4; ++i) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 26.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.td

       (IsInt8ElementType $weight),
       (IsConstTensor $weight),
       (IsInt32ElementType $matmul),
       (HasStaticShapeConstraint $weight)],
      [], (addBenefit 10)>;
    
    // Convert Matmul with hybrid inputs (f32 activation/int8 weight) to XlaDotV2
    def ConvertTFMatMulToXLADotV2OpWeightOnly : Pat<
      (TF_MatMulOp:$matmul
        $input,
        (TF_MulOp (TF_CastOp (TF_IdentityOp $weight), $truncate1), $scale),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Dec 10 05:52:02 UTC 2023
    - 21.1K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_test.cc

                                       "gradients/MatMul", false, true);
        TF_Operation* matmul2 = MatMul(expected_graph_, s_, const0, const3,
                                       "gradients/MatMul_1", true, false);
        expected_grad_outputs[0] = {matmul1, 0};
        expected_grad_outputs[1] = {matmul2, 0};
      }
    
      TF_Tensor* FloatTensor2x2(const float* values) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tests/end2end/unroll_batch_matmul_disabled.pbtxt

    # RUN: tf_tfl_translate -unfold_batchmatmul=false -tf-input-arrays=Placeholder,Placeholder_1 -tf-input-shapes=2,5,3:3,7 -tf-input-data-types=DT_FLOAT,DT_FLOAT -tf-output-arrays=MatMul -output-mlir %s -o - 2>&1 | FileCheck %s
    
    node {
      name: "Placeholder"
      op: "Placeholder"
      attr {
        key: "dtype"
        value {
          type: DT_FLOAT
        }
      }
      attr {
        key: "shape"
        value {
          shape {
            dim {
              size: 2
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 1.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_duplicate_resource_ops.mlir

        %outputs_7, %control_8 = tf_executor.island wraps "tf.Const"() {value = dense<"MatMul/b_0"> : tensor<1x!tf_type.string>} : () -> tensor<1x!tf_type.string>
        %outputs_9, %control_10 = tf_executor.island wraps "tf.VarHandleOp"() {container = "", shared_name = "MatMul/b_0"} : () -> tensor<!tf_type.resource<tensor<20x4096xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 26 04:26:16 UTC 2023
    - 10.5K bytes
    - Viewed (0)
  6. tensorflow/cc/framework/gradients_test.cc

          auto dv = Const(scope, {{1.0, 1.0}, {1.0, 1.0}});
          auto dt = MatMul(scope, dv, u, MatMul::TransposeB(true));
          auto du = MatMul(scope, t, dv, MatMul::TransposeA(true));
    
          auto dz = Const(scope, {{1.0, 1.0}, {1.0, 1.0}});
          auto dx = MatMul(scope, dz, y, MatMul::TransposeB(true));
          auto dy = MatMul(scope, x, dz, MatMul::TransposeA(true));
        } else {
          // Call AddSymbolicGradients.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 15 15:13:38 UTC 2023
    - 25K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

        }
        // FusedMatMul kernel does not support grad_a/grad_b attrs
        if ((matmul->hasAttr("grad_a") &&
             mlir::cast<BoolAttr>(matmul->getAttr("grad_a")).getValue()) ||
            (matmul->hasAttr("grad_b") &&
             mlir::cast<BoolAttr>(matmul->getAttr("grad_b")).getValue())) {
          (void)rewriter.notifyMatchFailure(matmul, [&](Diagnostic &diag) {
            diag << "FusedMatMul kernel does not support grad_a/grad_b attrs";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/tests/opens2s_gnmt_mixed_precision.golden_summary

     Transpose 2
     Unpack 1
    cluster 2 size 44
     Cast 2
     ConcatV2 2
     Const 18
     ExpandDims 1
     GatherV2 2
     Less 1
     MatMul 1
     Mul 1
     Pack 1
     Prod 2
     Range 1
     Reshape 3
     Shape 8
     StridedSlice 1
    cluster 3 size 10
     AddN 1
     Const 1
     MatMul 2
     Mul 1
     Reshape 3
     Sum 1
     Transpose 1
    cluster 4 size 11
     ConcatOffset 1
     Const 4
     ReverseSequence 1
     Slice 2
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jan 06 10:38:14 UTC 2023
    - 5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_weights.mlir

    // CHECK: %[[MATMUL:.*]] = "tf.MatMul"(%arg0, %[[DEQUANTIZED]]) <{transpose_a = false, transpose_b = false}> {attr_map = "0:transpose_a,1:transpose_a", device = ""} : (tensor<1x2x2x2xf32>, tensor<2x1024xf32>) -> tensor<*xf32>
    // CHECK: return %[[MATMUL]] : tensor<*xf32>
    
    // CHECK-LABEL: func.func private @composite_dequantize_uniform(%arg0: tensor<*xi8>) -> tensor<*xf32>
    // CHECK-DAG: %[[SCALE:.*]] = "tf.Const"() <{value = dense<0.0157480314> : tensor<f32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 42K bytes
    - Viewed (0)
  10. tensorflow/c/experimental/gradients/math_grad.cc

        std::string name_grad_B = "MatMul_Grad_B";
        if (!t_a && !t_b) {
          TF_RETURN_IF_ERROR(MatMul(ctx, upstream_grad, B.get(), &matmul_A_output,
                                    /*transpose_a = */ false,
                                    /*transpose_b = */ true, name_grad_A.c_str()));
    
          TF_RETURN_IF_ERROR(MatMul(ctx, A.get(), upstream_grad, &matmul_B_output,
                                    /*transpose_a = */ true,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 28 13:53:47 UTC 2024
    - 15.2K bytes
    - Viewed (0)
Back to top